init

app.py

@@ -136,12 +136,12 @@ def run(
     imgfile = os.path.join(opt.dataroot, fullname)
     lmfile = os.path.join(opt.lm_dir, name+'.txt')
     # 预处理数据
-    get_partmask.get_68lm(imgfile, lmfile, detector, predictor)
+    get_68lm(imgfile, lmfile, detector, predictor)
 
     imgs = []
     for part in ['eyel', 'eyer', 'nose', 'mouth']:
         savepath = os.path.join(opt.bg_dir + part, name+'.png')
-        get_partmask.get_partmask(imgfile, part, lmfile, savepath)
+        get_partmask(imgfile, part, lmfile, savepath)
         imgs.append(savepath)
 
     # data_loader = CreateDataLoader(opt)