From 692cfe7e8f88f95520e3486a01d4e09d9fe0b97a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Johannes=20Sch=C3=B6nberger?= Date: Thu, 8 Nov 2012 16:56:24 +0100 Subject: [PATCH] Use center of mass for peaks with equal accumulator values --- skimage/transform/hough_transform.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/skimage/transform/hough_transform.py b/skimage/transform/hough_transform.py index 3c27e346..79facb62 100644 --- a/skimage/transform/hough_transform.py +++ b/skimage/transform/hough_transform.py @@ -5,6 +5,7 @@ from itertools import izip as zip import numpy as np from scipy import ndimage from ._hough_transform import _probabilistic_hough +from skimage import measure, morphology def _hough(img, theta=None): @@ -208,7 +209,9 @@ def hough_peaks(hspace, angles, dists, min_distance=10, min_angle=10, hspace *= mask hspace_t = hspace > threshold - coords = np.transpose(hspace_t.nonzero()) + label_hspace = morphology.label(hspace_t) + props = measure.regionprops(label_hspace, ['Centroid']) + coords = np.array([np.round(p['Centroid']) for p in props], dtype=int) hspace_peaks = [] dist_peaks = []