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

Add missing support for num_peaks parameter

Browse Source
This commit is contained in:
Johannes Schönberger
2012-10-10 13:06:21 +02:00
parent d9a7bac999
commit ede5df6c70
2 changed files with 20 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/transform/hough_transform.py
+11 -1
View File
@@ -248,4 +248,14 @@ def hough_peaks(hspace, angles, dists, min_distance=10, min_angle=10,
dist_peaks.append(dists[dist_idx])
angle_peaks.append(angles[angle_idx])
return np.array(hspace_peaks), np.array(dist_peaks), np.array(angle_peaks)
hspace_peaks = np.array(hspace_peaks)
dist_peaks = np.array(dist_peaks)
angle_peaks = np.array(angle_peaks)
if num_peaks < len(hspace_peaks):
idx_maxsort = np.argsort(hspace_peaks)[::-1][:num_peaks]
hspace_peaks = hspace_peaks[idx_maxsort]
dist_peaks = dist_peaks[idx_maxsort]
angle_peaks = angle_peaks[idx_maxsort]
return hspace_peaks, dist_peaks, angle_peaks