wassname/scikit-image
1
0
Fork 0
mirror of https://github.com/wassname/scikit-image.git synced 2026-06-28 23:10:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix skimage.io.imread returns incorrect dimensions

Browse Source 
Fixes #1496
This commit is contained in:
Christoph Gohlke
2015-05-07 15:03:25 -07:00
parent c8a3df5521
commit 3c8bf02fa5
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/io/_io.py
+10 -2
View File
@@ -53,8 +53,16 @@ def imread(fname, as_grey=False, plugin=None, flatten=None,
with file_or_url_context(fname) as fname:
img = call_plugin('imread', fname, plugin=plugin, **plugin_args)
if as_grey and getattr(img, 'ndim', 0) >= 3:
img = rgb2grey(img)
if not hasattr(img, 'ndim'):
return img
if img.ndim > 2:
if img.shape[-1] not in (3, 4) and img.shape[-3] in (3, 4):
img = np.swapaxes(img, -1, -3)
img = np.swapaxes(img, -1, -2)
if as_grey:
img = rgb2grey(img)
return img