solve a bug

TachibanaYoshino · Aug 10, 2020 · 6f8a092 · 6f8a092
1 parent 4e55d27
commit 6f8a092
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tools/adjust_brightness.py b/tools/adjust_brightness.py
@@ -11,9 +11,9 @@ def read_img(image_path):
 # Calculates the average brightness in the specified irregular image
 def calculate_average_brightness(img):
     # Average value of three color channels
-    B = img[..., 0].mean()
+    R = img[..., 0].mean()
     G = img[..., 1].mean()
-    R = img[..., 2].mean()
+    B = img[..., 2].mean()
 
     brightness = 0.299 * R + 0.587 * G + 0.114 * B
     return brightness, B, G, R

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