fix extract

glucauze · Aug 3, 2023 · 26ac286 · 26ac286
1 parent be1cd15
commit 26ac286
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/client_api/faceswaplab_api_example.py b/client_api/faceswaplab_api_example.py
@@ -18,6 +18,8 @@
 
 address = "http://127.0.0.1:7860"
 
+# This has been tested on Linux platforms. This might requires some minor adaptations for windows.
+
 
 #############################
 # FaceSwap

diff --git a/scripts/faceswaplab_swapping/upscaled_inswapper.py b/scripts/faceswaplab_swapping/upscaled_inswapper.py
@@ -210,6 +210,7 @@ def compute_diff(bgr_fake: CV2ImgU8, aimg: CV2ImgU8) -> CV2ImgU8:
                     )
 
                     if options.improved_mask:
+                        logger.info("improved_mask")
                         mask = get_face_mask(aimg, bgr_fake)
                         bgr_fake = merge_images_with_mask(aimg, bgr_fake, mask)
 

diff --git a/scripts/faceswaplab_ui/faceswaplab_tab.py b/scripts/faceswaplab_ui/faceswaplab_tab.py
@@ -70,7 +70,9 @@ def extract_faces(
         logger.error("You need at least one image file to extract")
         return []
     try:
-        postprocess_options = PostProcessingOptions(*components)  # type: ignore
+        postprocess_options = dataclasses_from_flat_list(
+            [PostProcessingOptions], components
+        ).pop()
         images = [
             Image.open(file.name) for file in files
         ]  # potentially greedy but Image.open is supposed to be lazy