U ?hZ@sdZddlZddlZddlmZddlmZdZ dde DZ d Z e d de Dd d Z d dZ dddddZdddddZdS)z#Miscellaneous morphology functions.N)ndimage)warn)ZerosionZdilationZopeningclosingcCsi|]}|d|qS)Zgrey_.0xrrI/var/www/html/venv/lib/python3.8/site-packages/skimage/morphology/misc.py sr )Zbinary_erosionZbinary_dilationZbinary_openingZbinary_closingZ black_tophatZ white_tophatcCsi|] }||qSrrrrrr r scstdfdd }|S)aDecorator to add a default footprint 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 footprint of same dimension as the input image with connectivity 1. Ncs.|dkrt|jd}|f|d|i|S)N footprint)ndigenerate_binary_structurendim)imager argskwargsfuncrr func_out"sz#default_footprint..func_out)N) functoolswraps)rrrrr default_footprintsrcCs0|jtks,t|jtjs,td|jddS)Nz4Only bool or integer image types are supported. Got .)dtypeboolnpZ issubdtypeinteger TypeError)arrrr _check_dtype_supported+sr!@r outc Cst||dkr|}n ||dd<|dkr2|S|jtkrlt|j|}tj|tj d}tj |||dn|}zt | }Wnt k rt dYnXt|dkr|jtkrtd||k}||}d||<|S)apRemove objects smaller than the specified size. Expects ar to be an array with labeled objects, and removes objects smaller than min_size. If `ar` is bool, the image is first labeled. This leads to potentially different behavior for bool and 0-and-1 arrays. Parameters ---------- ar : ndarray (arbitrary shape, int or bool type) The array containing the objects of interest. If the array type is int, the ints must be non-negative. min_size : int, optional (default: 64) The smallest allowable object size. connectivity : int, {1, 2, ..., ar.ndim}, optional (default: 1) The connectivity defining the neighborhood of a pixel. Used during labelling if `ar` is bool. out : ndarray Array of the same shape as `ar`, into which the output is placed. By default, a new array is created. 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]]) >>> 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]]) >>> d = morphology.remove_small_objects(a, 6, out=a) >>> d is a True Nr)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?)r!copyrrrrrrZ zeros_likeZint32labelZbincountZravel ValueErrorlenr) r Zmin_size connectivityr$r ZccsZcomponent_sizesZ too_smallZtoo_small_maskrrr remove_small_objects2s*8   r+cCsvt||jtkrtdt|dk r8|jtkrFtdn|jtdd}tj||dt ||||d}tj||d|S)apRemove contiguous 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. out : ndarray Array of the same shape as `ar` and bool dtype, into which the output is placed. By default, a new array is created. 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]]) >>> 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]]) >>> d = morphology.remove_small_holes(a, 2, out=a) >>> 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?Nzout dtype must be boolT)r&r#) r!rrr UserWarningrZastyperZ logical_notr+)r Zarea_thresholdr*r$rrr remove_small_holess;   r-)r"r )r"r )__doc__numpyrrZscipyrrZ _shared.utilsrfuncsZskimage2ndimageupdaterr!r+r-rrrr s  [