diff --git a/skimage/segmentation/felzenszwalb.pyx b/skimage/segmentation/felzenszwalb.pyx index 8a927fa0..6b3fd90a 100644 --- a/skimage/segmentation/felzenszwalb.pyx +++ b/skimage/segmentation/felzenszwalb.pyx @@ -97,6 +97,7 @@ def felzenszwalb_segmentation(image, k, sigma=0.8): segment_size[seg_new] = segment_size[seg0] + segment_size[seg1] cint[seg_new] = cost # unravel the union find tree + flat = segments.ravel() old = np.zeros_like(flat) while (old != flat).any(): old = flat