feat: add argument to skip frames present in temp_path dir

2022-12-22 23:45:40 +01:00
parent dd1ecdd2a7
commit 45473fa490
9 changed files with 48 additions and 41 deletions
@@ -84,5 +84,5 @@ if __name__ == '__main__':
        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,\
-            no_simswaplogo=opt.no_simswaplogo,use_mask=opt.use_mask,crop_size=crop_size)
+            no_simswaplogo=opt.no_simswaplogo,use_mask=opt.use_mask,crop_size=crop_size,skip_existing_frames=opt.skip_existing_frames)