Merge pull request #317 from woctezuma/fix-rgb

Fix 3-channel images incorrectly read as if 1-channel

Author: carpedm20

utils.py

@@ -7,6 +7,7 @@
 import random
 import pprint
 import scipy.misc
+import cv2
 import numpy as np
 from time import gmtime, strftime
 from six.moves import xrange
@@ -36,7 +37,11 @@ def imread(path, grayscale = False):
   if (grayscale):
     return scipy.misc.imread(path, flatten = True).astype(np.float)
   else:
-    return scipy.misc.imread(path).astype(np.float)
+    # Reference: https://github.com/carpedm20/DCGAN-tensorflow/issues/162#issuecomment-315519747
+    img_bgr = cv2.imread(path)
+    # Reference: https://stackoverflow.com/a/15074748/
+    img_rgb = img_bgr[..., ::-1]
+    return img_rgb.astype(np.float)
 
 def merge_images(images, size):
   return inverse_transform(images)