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

change arguments of function estimate_transformation

Browse Source
This commit is contained in:
Johannes Schönberger
2012-07-15 15:47:12 +02:00
parent cb3c93a110
commit 2a75b78838
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/transform/geometric.py
+5 -2
View File
@@ -273,7 +273,7 @@ ESTIMATED_TRANSFORMATIONS = {
}
def estimate_transformation(ttype, *args, **kwargs):
def estimate_transformation(ttype, src, dst, order=None):
"""Estimate 2D geometric transformation parameters.
You can determine the over-, well- and under-determined parameters
@@ -322,7 +322,10 @@ def estimate_transformation(ttype, *args, **kwargs):
if ttype not in ESTIMATED_TRANSFORMATIONS:
raise NotImplemented('the transformation type \'%s\' is not'
'implemented' % ttype)
matrix = ESTIMATED_TRANSFORMATIONS[ttype][0](*args, **kwargs)
args = [src, dst]
if order is not None and ttype == 'polynomial':
args.append(order)
matrix = ESTIMATED_TRANSFORMATIONS[ttype][0](*args)
transform_func = ESTIMATED_TRANSFORMATIONS[ttype][1]
return GeometricTransformation(ttype, matrix, transform_func)