gavinyuan commited on
Commit
0540af1
1 Parent(s): 0fecb9a

update: PIPNet

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -187,6 +187,11 @@ def swap_image(
187
  G.eval()
188
  if gpu_mode:
189
  G = G.cuda()
 
 
 
 
 
190
  source_img = np.array(Image.open(source_image).convert("RGB"))
191
  net, detector = get_lmk_model()
192
  lmk = get_5_from_98(demo_image(source_img, net, detector)[0])
 
187
  G.eval()
188
  if gpu_mode:
189
  G = G.cuda()
190
+
191
+ os.system('ls -hlrt ./')
192
+ os.system('ls -hlrt ./third_party/PIPNet/FaceBoxesV2/utils/build')
193
+ os.system('cat -hlrt ./third_party/PIPNet/FaceBoxesV2/utils/nms/cpu_nms.pyx')
194
+
195
  source_img = np.array(Image.open(source_image).convert("RGB"))
196
  net, detector = get_lmk_model()
197
  lmk = get_5_from_98(demo_image(source_img, net, detector)[0])