From db92e07b7243c42fd61a00dbaf0625fc9970f28d Mon Sep 17 00:00:00 2001 From: Qing Date: Fri, 2 Dec 2022 13:40:35 +0800 Subject: [PATCH] change default IS model size to 384 --- lama_cleaner/interactive_seg.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lama_cleaner/interactive_seg.py b/lama_cleaner/interactive_seg.py index 94d7876..4470ffe 100644 --- a/lama_cleaner/interactive_seg.py +++ b/lama_cleaner/interactive_seg.py @@ -159,7 +159,7 @@ INTERACTIVE_SEG_MODEL_URL = os.environ.get( class InteractiveSeg: - def __init__(self, infer_size=448, open_kernel_size=3, dilate_kernel_size=3): + def __init__(self, infer_size=384, open_kernel_size=3, dilate_kernel_size=3): device = torch.device('cpu') model = load_jit_model(INTERACTIVE_SEG_MODEL_URL, device).eval() self.predictor = ISPredictor(model, device,