add sd_prevent_unmasked_area

2023-11-15 09:10:13 +08:00
parent f13bfb9085
commit 3e7a553273
4 changed files with 15 additions and 6 deletions
--- a/lama_cleaner/model/base.py
+++ b/lama_cleaner/model/base.py
@@ -67,8 +67,9 @@ class InpaintModel:

        result, image, mask = self.forward_post_process(result, image, mask, config)

-        mask = mask[:, :, np.newaxis]
-        result = result * (mask / 255) + image[:, :, ::-1] * (1 - (mask / 255))
+        if config.sd_prevent_unmasked_area:
+            mask = mask[:, :, np.newaxis]
+            result = result * (mask / 255) + image[:, :, ::-1] * (1 - (mask / 255))
        return result

    def forward_post_process(self, result, image, mask, config):