Use np.inf instead of float('inf')

This commit is contained in:
Johannes Schönberger
2015-03-08 19:37:53 -04:00
parent d21b6b8974
commit 47a4418140
2 changed files with 4 additions and 4 deletions
+3 -3
View File
@@ -505,18 +505,18 @@ def _dynamic_max_trials(n_inliers, n_samples, min_samples, probability):
Number of trials.
"""
if n_inliers == 0:
return float('inf')
return np.inf
nom = 1 - probability
if nom == 0:
return float('inf')
return np.inf
inlier_ratio = n_inliers / float(n_samples)
denom = 1 - inlier_ratio ** min_samples
if denom == 0:
return 1
elif denom == 1:
return float('inf')
return np.inf
nom = np.log(nom)
denom = np.log(denom)
+1 -1
View File
@@ -235,7 +235,7 @@ def test_ransac_dynamic_max_trials():
# e = 0%, min_samples = 5
assert_equal(_dynamic_max_trials(1, 100, 5, 0), 0)
assert_equal(_dynamic_max_trials(1, 100, 5, 1), float('inf'))
assert_equal(_dynamic_max_trials(1, 100, 5, 1), np.inf)
def test_ransac_invalid_input():