From 3ae1e3138adeef4cb6f6f1c87a9a87b46898bc1b Mon Sep 17 00:00:00 2001 From: "Gregory R. Lee" Date: Wed, 8 Jul 2015 14:41:34 -0400 Subject: [PATCH] BUG: bugfix for resize: upgrade image to 3d if necessary --- skimage/transform/_warps.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/skimage/transform/_warps.py b/skimage/transform/_warps.py index f069122a..54c1bc75 100644 --- a/skimage/transform/_warps.py +++ b/skimage/transform/_warps.py @@ -69,7 +69,9 @@ def resize(image, output_shape, order=1, mode='constant', cval=0, clip=True, if len(output_shape) == 3 and (image.ndim == 2 or output_shape[2] != image.shape[2]): dim = output_shape[2] - orig_dim = 1 if image.ndim == 2 else image.shape[2] + if image.ndim == 2: + image = image[:, :, np.newaxis] + orig_dim = image.shape[2] dim_scale = float(orig_dim) / dim map_rows, map_cols, map_dims = np.mgrid[:rows, :cols, :dim]