o :ήc !@sdZddlZddlZddlmZddlmZm Z ddl m Z dZ d d e DZ d Z e d d e Dd dZddZe dddddddddZe dddddddddZdS)z#Miscellaneous morphology functions.N)ndimage)warn remove_arg)_default_footprint)erosiondilationopeningclosingcCsi|]}|d|qS)grey_.0xr r >/tmp/pip-target-vg8gfxp4/lib/python/skimage/morphology/misc.py sr)binary_erosionbinary_dilationbinary_openingbinary_closing black_tophat white_tophatcCsi|]}||qSr r rr 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. Ncs,|dur t|j}|g|Rd|i|S)N footprint)rndim)imagerargskwargsfuncr rfunc_out#s z#default_footprint..func_out)N) functoolswraps)rr r rrdefault_footprintsr#cCs0|jtkst|jtjstd|jdSdS)Nz7Only bool or integer image types are supported. Got %s.)dtypeboolnp issubdtypeinteger TypeError)arr r r_check_dtype_supported,s r+in_placez1.0z Please use out argument instead.)changed_versionhelp_msg@Foutc Cst||dur d}|r|}n|dur|}n||dd<|dkr$|S|jtkrAt|j|}tj|tj d}tj |||dn|}z t | }Wn t yWt dwt|dkrg|jtkrgtd||k}||} d|| <|S) aCRemove 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. in_place : bool, optional (default: False) If ``True``, remove the objects in the input array itself. Otherwise, make a copy. Deprecated since version 0.19. Please use `out` instead. 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 NFr)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%ndigenerate_binary_structurerr& zeros_likeint32labelbincountravel ValueErrorlenr) r*min_size connectivityr,r1rccscomponent_sizes too_smalltoo_small_maskr r rremove_small_objects3s4?    rCcCst||jtkrtdt|dur|jtkrtdd}|r"|}n |dur-|jtdd}tj||dt ||||d}tj||d|S)aN Remove 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. in_place : bool, optional (default: False) If `True`, remove the connected components in the input array itself. Otherwise, make a copy. Deprecated since version 0.19. Please use `out` instead. 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 boolFT)r3r0) r+r$r%r UserWarningr)astyper& logical_notrC)r*area_thresholdr>r,r1r r rremove_small_holess"B  rH)r/rF)__doc__numpyr&r!scipyrr4 _shared.utilsrr footprintsrfuncsskimage2ndimageupdater#r+rCrHr r r rs0  f