diff --git a/test_video_swapmutil.py b/test_video_swapmutil.py index 26b9c81..de09bfc 100644 --- a/test_video_swapmutil.py +++ b/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) diff --git a/test_video_swapsingle.py b/test_video_swapsingle.py index 69f711e..b9cc525 100644 --- a/test_video_swapsingle.py +++ b/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)