gavinyuan commited on
Commit
82206b3
1 Parent(s): 4f18a93

update: mouth helper

Browse files
Files changed (1) hide show
  1. 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("./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,
 
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,