fix resize mask bug

2022-09-02 11:08:32 +08:00
parent 38c8837af7
commit 9ccc577be2
2 changed files with 2 additions and 3 deletions
--- 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: