Allow greyscale images in prediction.

tryolabs · Dec 4, 2017 · cf51a68 · cf51a68
1 parent 666e613
commit cf51a68
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/luminoth/utils/predicting.py b/luminoth/utils/predicting.py
@@ -36,7 +36,7 @@ def get_predictions(image_paths, config_files):
     for image_path in image_paths:
         with tf.gfile.Open(image_path, 'rb') as im_file:
             try:
-                image = Image.open(im_file)
+                image = Image.open(im_file).convert('RGB')
             except tf.errors.OutOfRangeError as e:
                 yield {
                     'error': '{}'.format(e),