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

ENH: use heapq to select the best match

Browse Source
This commit is contained in:
François Boulogne
2013-10-12 14:09:51 +02:00
parent 7e970b18cd
commit debd4d54d6
3 changed files with 35 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files
doc/examples/plot_circular_elliptical_hough_transform.py
+8 -7
View File
@@ -119,14 +119,15 @@ edges = filter.canny(image_gray, sigma=2.0,
# The accuracy corresponds to the bin size of a major axis.
# The value is chosen in order to get a single high accumulator.
# The threshold eliminates low accumulators
accum = hough_ellipse(edges, accuracy=10, threshold=170, min_size=50)
accum.sort(key=lambda x:x[5])
accum = hough_ellipse(edges, accuracy=10, threshold=150, min_size=50)
# Select the highest accumulator
best = heapq.nlargest(1, accum)[0]
# Estimated parameters for the ellipse
center_y = int(accum[-1][1])
center_x = int(accum[-1][2])
xradius = int(accum[-1][3])
yradius = int(accum[-1][4])
angle = np.pi - accum[-1][5]
center_y = int(best[1])
center_x = int(best[2])
xradius = int(best[3])
yradius = int(best[4])
angle = np.pi - best[5]
# Draw the ellipse on the original image
cx, cy = ellipse_perimeter(center_y, center_x,