solve a bug

TachibanaYoshino · Aug 10, 2020 · 858c80c · 858c80c
1 parent 6f8a092
commit 858c80c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()