o hZ@sdZddlZddlZddlmZddlmZdZ dde DZ d Z e d de Dd d Z d dZ dddddZdddddZdS)z#Miscellaneous morphology functions.N)ndimage)warn)erosiondilationopeningclosingcCsi|]}|d|qS)grey_.0xr r R/var/www/html/scripts/venv/lib/python3.10/site-packages/skimage/morphology/misc.py sr)binary_erosionbinary_dilationbinary_openingbinary_closing black_tophat white_tophatcCsi|]}||qSr r r r r rrscstdfdd }|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. Ncs0|dur t|jd}|g|Rd|i|S)N footprint)ndigenerate_binary_structurendim)imagerargskwargsfuncr rfunc_out"sz#default_footprint..func_out)N) functoolswraps)rr r rrdefault_footprintsr#cCs4|jtkst|jtjstd|jddSdS)Nz4Only bool or integer image types are supported. Got .)dtypeboolnp issubdtypeinteger TypeError)arr r r_check_dtype_supported+s  r,@routc Cst||dur |}n||dd<|dkr|S|jtkr6t|j|}tj|tj d}tj |||dn|}z t | }Wn t yLt dwt|dkr\|jtkr\td||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,copyr%r&rrrr' zeros_likeint32labelbincountravel ValueErrorlenr) r+min_size connectivityr/rccscomponent_sizes too_smalltoo_small_maskr r rremove_small_objects2s,8    r?cCsvt||jtkrtdt|dur|jtkrtdn|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)r1r.) r,r%r&r UserWarningr*astyper' logical_notr?)r+area_thresholdr:r/r r rremove_small_holess;  rD)r-r)__doc__numpyr'r!scipyrr _shared.utilsrfuncsskimage2ndimageupdater#r,r?rDr r r rs  [