diff --git a/lama_cleaner/model/base.py b/lama_cleaner/model/base.py index a287c3f..df769a6 100644 --- a/lama_cleaner/model/base.py +++ b/lama_cleaner/model/base.py @@ -93,7 +93,7 @@ class InpaintModel: (origin_size[1], origin_size[0]), interpolation=cv2.INTER_CUBIC) - original_pixel_indices = mask != 255 + original_pixel_indices = mask < 127 inpaint_result[original_pixel_indices] = image[:, :, ::-1][original_pixel_indices] if inpaint_result is None: diff --git a/lama_cleaner/model/fcf.py b/lama_cleaner/model/fcf.py index c05b1ed..ce52b21 100644 --- a/lama_cleaner/model/fcf.py +++ b/lama_cleaner/model/fcf.py @@ -1126,8 +1126,7 @@ class Generator(torch.nn.Module): FCF_MODEL_URL = os.environ.get( "FCF_MODEL_URL", - # "https://github.com/Sanster/models/releases/download/add_mat/Places_512_FullData_G.pth", - "/Users/qing/code/github/FcF-Inpainting/places_512_G.pth" + "https://github.com/Sanster/models/releases/download/add_fcf/places_512_G.pth", )