From 40ca4e68097589a6143a6754328a1f94786c118a Mon Sep 17 00:00:00 2001 From: tishibas67 Date: Wed, 18 Mar 2015 00:11:01 +0900 Subject: [PATCH] improved to load RGB image as grayscale image --- python/caffe/io.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/caffe/io.py b/python/caffe/io.py index 6ae2cf13cc0..acd8a1427c8 100644 --- a/python/caffe/io.py +++ b/python/caffe/io.py @@ -285,7 +285,7 @@ def load_image(filename, color=True): of size (H x W x 3) in RGB or of size (H x W x 1) in grayscale. """ - img = skimage.img_as_float(skimage.io.imread(filename)).astype(np.float32) + img = skimage.img_as_float(skimage.io.imread(filename, as_grey=not color)).astype(np.float32) if img.ndim == 2: img = img[:, :, np.newaxis] if color: