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

geometric_transform can transform single coordinate tuple

Browse Source
This commit is contained in:
Johannes Schönberger
2012-07-15 17:51:34 +02:00
parent d7b2c5b51b
commit afb479d766
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/transform/geometric.py
+10 -1
View File
@@ -43,13 +43,22 @@ def geometric_transform(coords, matrix):
coords : Nx2 array
transformed coordinates
"""
coords = np.asarray(coords)
shape = coords.shape
if shape == (2,):
coords = np.array([coords])
x, y = np.transpose(coords)
src = np.vstack((x, y, np.ones_like(x)))
dst = np.dot(src.transpose(), matrix.transpose())
# rescale to homogeneous coordinates
dst[:, 0] /= dst[:, 2]
dst[:, 1] /= dst[:, 2]
return dst[:, :2]
if shape == (2,):
return dst[0, :2]
else:
return dst[:, :2]
class GeometricTransformation(object):