浏览代码

Script for video face swapping(You can choose to add or not add a watermark )

NNNNAI 4 年之前
父节点
当前提交
d2af8165f2
共有 2 个文件被更改,包括 2 次插入2 次删除
  1. 1 1
      test_video_swapmutil.py
  2. 1 1
      test_video_swapsingle.py

+ 1 - 1
test_video_swapmutil.py

@@ -68,5 +68,5 @@ if __name__ == '__main__':
     latend_id = model.netArc(img_id_downsample)
     latend_id = F.normalize(latend_id, p=2, dim=1)
 
-    video_swap(opt.video_path, latend_id, model, app, opt.output_path,temp_results_dir=opt.temp_path)
+    video_swap(opt.video_path, latend_id, model, app, opt.output_path,temp_results_dir=opt.temp_path,no_simswaplogo=opt.no_simswaplogo)
 

+ 1 - 1
test_video_swapsingle.py

@@ -68,5 +68,5 @@ if __name__ == '__main__':
     latend_id = model.netArc(img_id_downsample)
     latend_id = F.normalize(latend_id, p=2, dim=1)
 
-    video_swap(opt.video_path, latend_id, model, app, opt.output_path,temp_results_dir=opt.temp_path)
+    video_swap(opt.video_path, latend_id, model, app, opt.output_path,temp_results_dir=opt.temp_path,no_simswaplogo=opt.no_simswaplogo)