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

Merge pull request #1 from jni/peak-local-max-indices-nd

Browse Source 
Make `exclude_borders` option in peak_local_max nD
This commit is contained in:
yangzetian
2013-04-03 03:03:08 -07:00
parent afde2b8ab8 6e4ec78de2
commit 4a4517cb06
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 range(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)