瀏覽代碼

add swap face test

ming 3 年之前
父節點
當前提交
3bb5c9600e
共有 2 個文件被更改,包括 3 次插入2 次删除
  1. 2 1
      main.py
  2. 1 1
      util/swap_face.py

+ 2 - 1
main.py

@@ -113,7 +113,8 @@ async def progress_page():
 async def swapFace(req:swap_req):
     sf = swap_face(req.imgurl)
     result = sf.run()
-    notify_group(result)
+    #notify_group(result)
+    print(result)
     
 
 @app.post("/make_anchor_video_v2")

+ 1 - 1
util/swap_face.py

@@ -18,7 +18,7 @@ class swap_face():
             return '圖片錯誤'
         
 
-        os.system('python3 FaceSwap/main_video.py --src_img '+src_img+' --video_path nina_noGesture_adj.mp4 --correct_color --save_path '+sv_path)    
+        os.system('python3 FaceSwap/main_video.py --src_img '+src_img+' --video_path FaceSwap/nina_noGesture_adj.mp4 --correct_color --save_path '+sv_path)    
         while True:
             print('waiting...')
             if os.path.exists(sv_path):