Spaces:
Running
Running
gavinyuan
commited on
Commit
•
0540af1
1
Parent(s):
0fecb9a
update: PIPNet
Browse files
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])
|