Spaces:
Running
Running
gavinyuan
commited on
Commit
•
82206b3
1
Parent(s):
4f18a93
update: mouth helper
Browse files- inference/tricks.py +2 -2
inference/tricks.py
CHANGED
@@ -104,9 +104,9 @@ class Trick(object):
|
|
104 |
# benchmark=None,
|
105 |
# demo_folder=None,
|
106 |
# )
|
107 |
-
pt_path = make_abs_path("
|
108 |
self.mouth_helper = FSGenerator(
|
109 |
-
make_abs_path("
|
110 |
mouth_net_param={"use": False},
|
111 |
in_size=256,
|
112 |
downup=False,
|
|
|
104 |
# benchmark=None,
|
105 |
# demo_folder=None,
|
106 |
# )
|
107 |
+
pt_path = make_abs_path("../weights/extracted/G_t34_helper_post.pth")
|
108 |
self.mouth_helper = FSGenerator(
|
109 |
+
make_abs_path("../weights/arcface/ms1mv3_arcface_r100_fp16/backbone.pth"),
|
110 |
mouth_net_param={"use": False},
|
111 |
in_size=256,
|
112 |
downup=False,
|