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

Add support for 2(+1)D images and given output_shape for 2D images

Browse Source
This commit is contained in:
Johannes Schönberger
2014-09-15 23:34:33 -04:00
parent f9009410ba
commit 9b4f2d9032
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/transform/_geometric.py
+7
View File
@@ -1180,6 +1180,13 @@ def warp(image, inverse_map=None, map_args={}, output_shape=None, order=1,
def coord_map(*args):
return inverse_map(*args, **map_args)
# Input image is 2D and has color channel, but output_shape is
# given for 2-D images. Automatically add the color channel
# dimensionality.
if len(input_shape) == 3 and len(output_shape) == 2:
output_shape = (output_shape[0], output_shape[1],
input_shape[2])
coords = warp_coords(coord_map, output_shape)
# Pre-filtering not necessary for order 0, 1 interpolation