Jelajahi Sumber

add swap face test

ming 3 tahun lalu
induk
melakukan
3bb5c9600e
2 mengubah file dengan 3 tambahan dan 2 penghapusan
  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):
 async def swapFace(req:swap_req):
     sf = swap_face(req.imgurl)
     sf = swap_face(req.imgurl)
     result = sf.run()
     result = sf.run()
-    notify_group(result)
+    #notify_group(result)
+    print(result)
     
     
 
 
 @app.post("/make_anchor_video_v2")
 @app.post("/make_anchor_video_v2")

+ 1 - 1
util/swap_face.py

@@ -18,7 +18,7 @@ class swap_face():
             return '圖片錯誤'
             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:
         while True:
             print('waiting...')
             print('waiting...')
             if os.path.exists(sv_path):
             if os.path.exists(sv_path):