B \0@sdZddlZddlZddlmZddlmZddl m Z dZ e d d e DZ d Z e e d d e Dd dZddZdddZdddZdS)z#Miscellaneous morphology functions.N)ndimage)warn)_default_selem)ZerosionZdilationZopeningclosingccs|]}|d|fVqdS)Zgrey_N).0xrr6lib/python3.7/site-packages/skimage/morphology/misc.py sr )Zbinary_erosionZbinary_dilationZbinary_openingZbinary_closingZ black_tophatZ white_tophatccs|]}||fVqdS)Nr)r r rrr r scstdfdd }|S)aDecorator to add a default structuring element to morphology functions. Parameters ---------- func : function A morphology function such as erosion, dilation, opening, closing, white_tophat, or black_tophat. Returns ------- func_out : function The function, using a default structuring element of same dimension as the input image with connectivity 1. Ncs*|dkrt|j}|f|d|i|S)Nselem)rndim)Zimager argskwargs)funcrr func_out#s zdefault_selem..func_out)N) functoolswraps)rrr)rr default_selemsrcCs,|jtks(t|jtjs(td|jdS)Nz7Only bool or integer image types are supported. Got %s.)dtypeboolnpZ issubdtypeZinteger TypeError)arrrr _check_dtype_supported,sr@Fc Cst||r|}n|}|dkr&|S|jtkr`t|j|}tj|tj d}tj |||dn|}yt | }Wnt k rt dYnXt|dkr|jtkrtd||k}||} d|| <|S)aRemove connected components smaller than the specified size. Parameters ---------- ar : ndarray (arbitrary shape, int or bool type) The array containing the connected components of interest. If the array type is int, it is assumed that it contains already-labeled objects. The ints must be non-negative. min_size : int, optional (default: 64) The smallest allowable connected component size. connectivity : int, {1, 2, ..., ar.ndim}, optional (default: 1) The connectivity defining the neighborhood of a pixel. in_place : bool, optional (default: False) If `True`, remove the connected components in the input array itself. Otherwise, make a copy. Raises ------ TypeError If the input array is of an invalid type, such as float or string. ValueError If the input array contains negative values. Returns ------- out : ndarray, same shape and type as input `ar` The input array with small connected components removed. Examples -------- >>> from skimage import morphology >>> a = np.array([[0, 0, 0, 1, 0], ... [1, 1, 1, 0, 0], ... [1, 1, 1, 0, 1]], bool) >>> b = morphology.remove_small_objects(a, 6) >>> b array([[False, False, False, False, False], [ True, True, True, False, False], [ True, True, True, False, False]], dtype=bool) >>> c = morphology.remove_small_objects(a, 7, connectivity=2) >>> c array([[False, False, False, True, False], [ True, True, True, False, False], [ True, True, True, False, False]], dtype=bool) >>> d = morphology.remove_small_objects(a, 6, in_place=True) >>> d is a True r)r)outputz{Negative value labels are not supported. Try relabeling the input with `scipy.ndimage.label` or `skimage.morphology.label`.rz[Only one label was provided to `remove_small_objects`. Did you mean to use a boolean array?)rcopyrrndiZgenerate_binary_structurerrZ zeros_likeZint32ZlabelZbincountZravel ValueErrorlenr) rmin_size connectivityin_placeoutr ZccsZcomponent_sizesZ too_smallZtoo_small_maskrrr remove_small_objects3s*3 r&cCst||jtkrtdt|dk r0td|}|r:|}n|}|rTt||}n t|}t||||}|r~t||}n t|}|S)aRemove continguous holes smaller than the specified size. Parameters ---------- ar : ndarray (arbitrary shape, int or bool type) The array containing the connected components of interest. area_threshold : int, optional (default: 64) The maximum area, in pixels, of a contiguous hole that will be filled. Replaces `min_size`. connectivity : int, {1, 2, ..., ar.ndim}, optional (default: 1) The connectivity defining the neighborhood of a pixel. in_place : bool, optional (default: False) If `True`, remove the connected components in the input array itself. Otherwise, make a copy. Raises ------ TypeError If the input array is of an invalid type, such as float or string. ValueError If the input array contains negative values. Returns ------- out : ndarray, same shape and type as input `ar` The input array with small holes within connected components removed. Examples -------- >>> from skimage import morphology >>> a = np.array([[1, 1, 1, 1, 1, 0], ... [1, 1, 1, 0, 1, 0], ... [1, 0, 0, 1, 1, 0], ... [1, 1, 1, 1, 1, 0]], bool) >>> b = morphology.remove_small_holes(a, 2) >>> b array([[ True, True, True, True, True, False], [ True, True, True, True, True, False], [ True, False, False, True, True, False], [ True, True, True, True, True, False]], dtype=bool) >>> c = morphology.remove_small_holes(a, 2, connectivity=2) >>> c array([[ True, True, True, True, True, False], [ True, True, True, False, True, False], [ True, False, False, True, True, False], [ True, True, True, True, True, False]], dtype=bool) >>> d = morphology.remove_small_holes(a, 2, in_place=True) >>> d is a True Notes ----- If the array type is int, it is assumed that it contains already-labeled objects. The labels are not kept in the output image (this function always outputs a bool image). It is suggested that labeling is completed after using this function. z\Any labeled images will be returned as a boolean array. Did you mean to use a boolean array?Nz\the min_size argument is deprecated and will be removed in 0.16. Use area_threshold instead.) rrrr UserWarningrrZ logical_notr&)rZarea_thresholdr#r$r"r%rrr remove_small_holess$=   r()rrF)rrFN)__doc__ZnumpyrrZscipyrrZ _shared.utilsrr rZfuncsdictZskimage2ndimageupdaterrr&r(rrrr s    V