fix bug in ransac

[imago.git] / src / ransac.py

diff --git a/src/ransac.py b/src/ransac.py
index 5d5d218..4908421 100644 (file)
--- a/src/ransac.py
+++ b/src/ransac.py
@@ -57,10 +57,9 @@ def iterate(model, distance):
         score = new_score
         try:
             estimate = model.get(consensual)
         score = new_score
         try:
             estimate = model.get(consensual)

+            new_score, consensual = model.score(estimate, distance)

         except NP.linalg.LinAlgError:
             pass
         except NP.linalg.LinAlgError:
             pass

-        estimate = model.get(consensual)
-        new_score, consensual = model.score(estimate, distance)

     return score, estimate, consensual
         
 def estimate(data, dist, k, modelClass=Linear_model):
     return score, estimate, consensual
         
 def estimate(data, dist, k, modelClass=Linear_model):