Merge pull request brade31919#36 from omegacoleman/master

Fix ValueError while using non 3-channeled PNG
dinavod12 · Jun 15, 2018 · feb5a1d · feb5a1d
2 parents 2e3766a + c5ab3f6
commit feb5a1d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/model.py b/lib/model.py
@@ -153,7 +153,7 @@ def test_data_loader(FLAGS):
 
     # Read in and preprocess the images
     def preprocess_test(name, mode):
-        im = sic.imread(name).astype(np.float32)
+        im = sic.imread(name, mode="RGB").astype(np.float32)
         # check grayscale image
         if im.shape[-1] != 3:
             h, w = im.shape
@@ -196,7 +196,7 @@ def inference_data_loader(FLAGS):
 
     # Read in and preprocess the images
     def preprocess_test(name):
-        im = sic.imread(name).astype(np.float32)
+        im = sic.imread(name, mode="RGB").astype(np.float32)
         # check grayscale image
         if im.shape[-1] != 3:
             h, w = im.shape