Set constant value for maximum filter explicitly

This commit is contained in:
Johannes Schönberger
2012-10-10 13:05:24 +02:00
parent ede5df6c70
commit ee098224eb
+2 -2
View File
@@ -202,9 +202,9 @@ def hough_peaks(hspace, angles, dists, min_distance=10, min_angle=10,
distance_size = 2 * min_distance + 1
angle_size = 2 * min_angle + 1
hspace_max = ndimage.maximum_filter1d(hspace, size=distance_size, axis=0,
mode='constant')
mode='constant', cval=0)
hspace_max = ndimage.maximum_filter1d(hspace_max, size=angle_size, axis=1,
mode='constant')
mode='constant', cval=0)
mask = (hspace == hspace_max)
hspace *= mask
hspace_t = hspace > threshold