From 048c6c06ebe9935aa6844a09ed85396c156befbd Mon Sep 17 00:00:00 2001 From: Tony S Yu Date: Tue, 15 May 2012 15:56:26 -0400 Subject: [PATCH] BUG: Fix stackcopy for grayscale image. --- skimage/transform/_warp.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/skimage/transform/_warp.py b/skimage/transform/_warp.py index 29a8531e..6477c988 100644 --- a/skimage/transform/_warp.py +++ b/skimage/transform/_warp.py @@ -22,7 +22,11 @@ def _stackcopy(a, b): Color images are stored as an ``MxNx3`` or ``MxNx4`` arrays. """ - a[:] = b[:, :, np.newaxis] + if a.ndim == 3: + a[:] = b[:, :, np.newaxis] + else: + a[:] = b + def warp(image, reverse_map, map_args={}, output_shape=None, order=1, mode='constant', cval=0.):