mirror of
https://github.com/wassname/scikit-image.git
synced 2026-07-11 06:06:23 +08:00
Use np.inf instead of float('inf')
This commit is contained in:
@@ -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)
|
||||
|
||||
@@ -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():
|
||||
|
||||
Reference in New Issue
Block a user