|
@@ -18,7 +18,7 @@ def _totensor(array):
|
|
|
img = tensor.transpose(0, 1).transpose(0, 2).contiguous()
|
|
|
return img.float().div(255)
|
|
|
|
|
|
-def video_swap(video_path, id_vetor, swap_model, detect_model, save_path, temp_results_dir='./temp_results', crop_size=224):
|
|
|
+def video_swap(video_path, id_vetor, swap_model, detect_model, save_path, temp_results_dir='./temp_results', crop_size=224, no_simswaplogo = False):
|
|
|
video_forcheck = VideoFileClip(video_path)
|
|
|
if video_forcheck.audio is None:
|
|
|
no_audio = True
|
|
@@ -71,7 +71,7 @@ def video_swap(video_path, id_vetor, swap_model, detect_model, save_path, temp_r
|
|
|
|
|
|
|
|
|
|
|
|
- reverse2wholeimage(swap_result_list, frame_mat_list, crop_size, frame, logoclass,os.path.join(temp_results_dir, 'frame_{:0>7d}.jpg'.format(frame_index)))
|
|
|
+ reverse2wholeimage(swap_result_list, frame_mat_list, crop_size, frame, logoclass,os.path.join(temp_results_dir, 'frame_{:0>7d}.jpg'.format(frame_index)),no_simswaplogo)
|
|
|
|
|
|
else:
|
|
|
if not os.path.exists(temp_results_dir):
|