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

Make exclude_borders option in peak_local_max nD

Browse Source 
Prior modifications to peak_local_max make it nd-aware, but this does
not extend to the exclude_borders option. This commit changes that.
This commit is contained in:
Juan Nunez-Iglesias
2013-04-03 17:56:04 +11:00
parent afde2b8ab8
commit d9e6b509c5
1 changed files with 6 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
skimage/feature/peak.py
+6 -5
View File
@@ -130,11 +130,12 @@ def peak_local_max(image, min_distance=10, threshold_abs=0, threshold_rel=0.1,
image *= mask
if exclude_border:
# Remove the image borders
image[:min_distance] = 0
image[-min_distance:] = 0
image[:, :min_distance] = 0
image[:, -min_distance:] = 0
# zero out the image borders
for i in xrange(image.ndim):
image = image.swapaxes(0, i)
image[:min_distance] = 0
image[-min_distance:] = 0
image = image.swapaxes(0, i)
# find top peak candidates above a threshold
peak_threshold = max(np.max(image.ravel()) * threshold_rel, threshold_abs)