From 858c80c9d9df14395aa08d4efe909eb2fbb9a359 Mon Sep 17 00:00:00 2001
From: Alice <ghj9527@163.com>
Date: Mon, 10 Aug 2020 23:48:41 +0800
Subject: [PATCH] solve a bug

---
 video2anime.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/video2anime.py b/video2anime.py
index 00bd98b..aacaafc 100644
--- a/video2anime.py
+++ b/video2anime.py
@@ -116,7 +116,7 @@ def cvt2anime_video(video, output, checkpoint_dir, output_format='MP4V', img_siz
             fake_img = sess.run(test_generated, feed_dict={test_real: img})
             fake_img = inverse_image(fake_img)
             fake_img = adjust_brightness_from_src_to_dst(fake_img, cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
-            out.write(fake_img)
+            out.write(cv2.cvtColor(fake_img, cv2.COLOR_BGR2RGB))
             pbar.update(1)
 
         pbar.close()