From eb66e343514b891ea34b5e5236ae6eb0f5cbca40 Mon Sep 17 00:00:00 2001 From: Stefan van der Walt Date: Fri, 28 Jun 2013 16:19:49 -0500 Subject: [PATCH] Fix label casting for numpy 1.7. --- skimage/morphology/misc.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/skimage/morphology/misc.py b/skimage/morphology/misc.py index 820ce4a1..4e71b224 100644 --- a/skimage/morphology/misc.py +++ b/skimage/morphology/misc.py @@ -66,7 +66,8 @@ def remove_small_objects(ar, min_size=64, connectivity=1, in_place=False): if out.dtype == bool: selem = nd.generate_binary_structure(ar.ndim, connectivity) - ccs = nd.label(ar, selem)[0] + ccs = np.zeros_like(ar, dtype=np.int) + nd.label(ar, selem, output=ccs) else: ccs = out