浏览代码

Some related scripts for face swapping.

NNNNAI 4 年之前
父节点
当前提交
dbcfc9a44b
共有 1 个文件被更改,包括 1 次插入0 次删除
  1. 1 0
      util/videoswap.py

+ 1 - 0
util/videoswap.py

@@ -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