From dbcfc9a44b3298288460f9611c623612f7d0975d Mon Sep 17 00:00:00 2001
From: NNNNAI <844294823@qq.com>
Date: Sun, 20 Jun 2021 16:28:06 +0800
Subject: [PATCH] Some related scripts for face swapping.

---
 util/videoswap.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/util/videoswap.py b/util/videoswap.py
index a19cecc..064c2d0 100644
--- a/util/videoswap.py
+++ b/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