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):