CalvinBackup/neuralchen-SimSwap
1
0
Fork 0
Code Issues 288 Pull Requests 16 Actions Packages Projects Releases 2 Wiki Activity

Update videoswap.py

Browse Source
This commit is contained in:
NNNNAI
2021-07-02 19:17:01 +08:00
parent 1faf9fd496
commit c5bc396123
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
util/videoswap.py
+2 -2
View File
@@ -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):