Some related scripts for face swapping.

2021-06-20 16:28:06 +08:00
parent 65905e476d
commit dbcfc9a44b
1 changed files with 1 additions and 0 deletions
@@ -66,6 +66,7 @@ def video_swap(video_path, id_vetor, swap_model, detect_model, save_path, temp_r
            else:
                if not os.path.exists(temp_results_dir):
                    os.mkdir(temp_results_dir)
+                frame = logoclass.apply_frames(frame.astype(np.uint8))
                cv2.imwrite(os.path.join(temp_results_dir, 'frame_{:0>7d}.jpg'.format(frame_index)), frame)
        else:
            break