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

fix num_peaks parameter bug in peak_local_max

Browse Source
This commit is contained in:
Johannes Schönberger
2012-08-10 20:45:50 +02:00
parent 8b544f2f69
commit a87779b650
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/feature/peak.py
+3 -3
View File
@@ -95,9 +95,9 @@ def peak_local_max(image, min_distance=10, threshold='deprecated',
# get coordinates of peaks
coordinates = np.transpose(image_t.nonzero())
if len(coordinates) > num_peaks:
intensities = image[tuple(coordinates.T)]
if coordinates.shape[0] > num_peaks:
intensities = image[coordinates[:, 0], coordinates[:, 1]]
idx_maxsort = np.argsort(intensities)[::-1]
coordinates = coordinates[idx_maxsort][:2]
coordinates = coordinates[idx_maxsort][:num_peaks]
return coordinates