wassname/scikit-image
1
0
Fork 0
mirror of https://github.com/wassname/scikit-image.git synced 2026-07-08 14:56:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix bug with 2D images

Browse Source
This commit is contained in:
Neil
2011-07-16 18:44:25 +02:00
parent 0a71c9060e
commit 3f12aef7ed
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scikits/image/io/_plugins/util.py
+5 -4
View File
@@ -135,14 +135,15 @@ def prepare_for_display(npy_img):
if npy_img.ndim == 2 or \
(npy_img.ndim == 3 and npy_img.shape[2] == 1):
npy_plane = npy_img.reshape((height, width))
if np.issubdtype(npy_img.dtype, float):
out[:,:,0] = npy_img*255
out[:,:,0] = npy_plane*255
out[:,:,1] = out[:,:,0]
out[:,:,2] = out[:,:,0]
else:
out[:,:,0] = npy_img
out[:,:,1] = npy_img
out[:,:,2] = npy_img
out[:,:,0] = npy_plane
out[:,:,1] = npy_plane
out[:,:,2] = npy_plane
elif npy_img.ndim == 3:
if npy_img.shape[2] == 3 or npy_img.shape[2] == 4: