update: mouth helper

inference/tricks.py

@@ -97,14 +97,22 @@ class Trick(object):
         return helper_face * i_r_mouth_mask + i_r * (1 - i_r_mouth_mask)
 
     def load_mouth_helper(self):
-        from inference.ffplus.eval import EvaluatorFaceShifter
-        mouth_helper_pl = EvaluatorFaceShifter(
-            load_path="/apdcephfs/share_1290939/gavinyuan/out/triplet10w_34/epoch=13-step=737999.ckpt",
-            pt_path=make_abs_path("../ffplus/extracted_ckpt/G_t34_helper_post.pth"),
-            benchmark=None,
-            demo_folder=None,
+        from modules.networks.faceshifter import FSGenerator
+        # mouth_helper_pl = EvaluatorFaceShifter(
+        #     load_path="/apdcephfs/share_1290939/gavinyuan/out/triplet10w_34/epoch=13-step=737999.ckpt",
+        #     pt_path=make_abs_path("../ffplus/extracted_ckpt/G_t34_helper_post.pth"),
+        #     benchmark=None,
+        #     demo_folder=None,
+        # )
+        pt_path = make_abs_path("./weights/extracted/G_t34_helper_post.pth")
+        self.mouth_helper = FSGenerator(
+            make_abs_path("./weights/arcface/ms1mv3_arcface_r100_fp16/backbone.pth"),
+            mouth_net_param={"use": False},
+            in_size=256,
+            downup=False,
         )
-        self.mouth_helper = mouth_helper_pl.faceswap_model
+        self.mouth_helper.load_state_dict(torch.load(pt_path, "cpu"), strict=True)
+        self.mouth_helper.eval()
         print("[Mouth helper] loaded.")
 
 

weights/extracted/G_t34_helper_post.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0192f044350b8020c45251d7f2e80ecbc39ec32b4b8129f142cee76f1b917458
+size 998751980