clean code

2023-11-15 08:58:52 +08:00
parent 58b931fdb2
commit f13bfb9085
4 changed files with 9 additions and 26 deletions
--- a/lama_cleaner/model/sdxl.py
+++ b/lama_cleaner/model/sdxl.py
@@ -85,14 +85,6 @@ class SDXL(DiffusionInpaintModel):
        output = cv2.cvtColor(output, cv2.COLOR_RGB2BGR)
        return output

-    def forward_post_process(self, result, image, mask, config):
-        if config.sd_match_histograms:
-            result = self._match_histograms(result, image[:, :, ::-1], mask)
-
-        if config.sd_mask_blur != 0:
-            k = 2 * config.sd_mask_blur + 1
-            mask = cv2.GaussianBlur(mask, (k, k), 0)
-        return result, image, mask

    @staticmethod
    def is_downloaded() -> bool: