Convert all loaded images to RGB.

This makes the code compatible with greyscale data. Addresses comment in #8.
MarvinTeichmann · Feb 17, 2017 · f7fdb24 · f7fdb24
1 parent 3927ad3
commit f7fdb24
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/evals/kitti_eval.py b/evals/kitti_eval.py
@@ -69,8 +69,8 @@ def evaluate(hypes, sess, image_pl, inf_out):
                 image_file = os.path.join(image_dir, image_file)
                 gt_file = os.path.join(image_dir, gt_file)
 
-                image = scp.misc.imread(image_file)
-                gt_image = scp.misc.imread(gt_file)
+                image = scp.misc.imread(image_file, mode='RGB')
+                gt_image = scp.misc.imread(gt_file, mode='RGB')
 
                 if hypes['jitter']['fix_shape']:
                     shape = image.shape

diff --git a/inputs/kitti_seg_input.py b/inputs/kitti_seg_input.py
@@ -114,7 +114,7 @@ def _load_gt_file(hypes, data_file=None):
             gt_image_file = os.path.join(base_path, gt_image_file)
             assert os.path.exists(gt_image_file), \
                 "File does not exist: %s" % gt_image_file
-            image = scipy.misc.imread(image_file)
+            image = scipy.misc.imread(image_file, mode='RGB')
             # Please update Scipy, if mode='RGB' is not avaible
             gt_image = scp.misc.imread(gt_image_file, mode='RGB')