|
@@ -76,7 +76,9 @@ def video_swap(video_path, id_vetor, swap_model, detect_model, save_path, temp_r
|
|
else:
|
|
else:
|
|
if not os.path.exists(temp_results_dir):
|
|
if not os.path.exists(temp_results_dir):
|
|
os.mkdir(temp_results_dir)
|
|
os.mkdir(temp_results_dir)
|
|
- frame = logoclass.apply_frames(frame.astype(np.uint8))
|
|
|
|
|
|
+ frame = frame.astype(np.uint8)
|
|
|
|
+ if not no_simswaplogo:
|
|
|
|
+ frame = logoclass.apply_frames(frame)
|
|
cv2.imwrite(os.path.join(temp_results_dir, 'frame_{:0>7d}.jpg'.format(frame_index)), frame)
|
|
cv2.imwrite(os.path.join(temp_results_dir, 'frame_{:0>7d}.jpg'.format(frame_index)), frame)
|
|
else:
|
|
else:
|
|
break
|
|
break
|