kamp/IOPaint
1
0
Fork 0
Code Issues Pull Requests Releases 1 Activity

change default IS model size to 384

Browse Source
This commit is contained in:
Qing 2022-12-02 13:40:35 +08:00
parent 2611d2fb44
commit db92e07b72
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lama_cleaner/interactive_seg.py
View File

@ -159,7 +159,7 @@ INTERACTIVE_SEG_MODEL_URL = os.environ.get(
class InteractiveSeg: 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') device = torch.device('cpu')
model = load_jit_model(INTERACTIVE_SEG_MODEL_URL, device).eval() model = load_jit_model(INTERACTIVE_SEG_MODEL_URL, device).eval()
self.predictor = ISPredictor(model, device, self.predictor = ISPredictor(model, device,