From c5bc396123641dc02240c28b67ed7dcd3645b972 Mon Sep 17 00:00:00 2001 From: NNNNAI <844294823@qq.com> Date: Fri, 2 Jul 2021 19:17:01 +0800 Subject: [PATCH] Update videoswap.py --- util/videoswap.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/util/videoswap.py b/util/videoswap.py index 11e60c3..acc4aca 100644 --- a/util/videoswap.py +++ b/util/videoswap.py @@ -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):