add anime segmentation

2023-05-09 19:07:12 +08:00
parent 7fcce78e40
commit e5ac6a105a
11 changed files with 510 additions and 5 deletions
--- a/lama_cleaner/plugins/remove_bg.py
+++ b/lama_cleaner/plugins/remove_bg.py
@@ -28,7 +28,7 @@ class RemoveBG(BasePlugin):

        # return BGRA image
        output = remove(bgr_np_img, session=self.session)
-        return output
+        return cv2.cvtColor(output, cv2.COLOR_BGRA2RGBA)

    def check_dep(self):
        try: