From a87779b650e4fa1e6abd08510d2dc241e6fd9dbd Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Johannes=20Sch=C3=B6nberger?= Date: Fri, 10 Aug 2012 20:45:50 +0200 Subject: [PATCH] fix num_peaks parameter bug in peak_local_max --- skimage/feature/peak.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/skimage/feature/peak.py b/skimage/feature/peak.py index d8e54dd6..57eb3bb7 100644 --- a/skimage/feature/peak.py +++ b/skimage/feature/peak.py @@ -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