o :ήc8@sddlZddlZddlmZddlmZddlmZddl m Z ddZ d d Z ej d d   dddddZej d d   dddddZej dd   dddddZej dd   dddddZdS)N)utils)gaussian)convert_to_float)resizecCsJt|}|dur||j}|f|jd}nd}t||||||d|S)z?Return image with each channel smoothed by the Gaussian filter.N)outputmodecval channel_axis)np empty_likendimr)imagesigmar r r smoothedrA/tmp/pip-target-vg8gfxp4/lib/python/skimage/transform/pyramids.py_smooth s   rcCs|dkrtddS)Nrz#scale factor must be greater than 1) ValueError)factorrrr _check_factorsr)multichannel_positionrreflectFr c stt||}dur"|jtfddt|jD} n tfdd|jD} |dur8dd}t||||} t| | |||dd} | S) aSmooth and then downsample image. Parameters ---------- image : ndarray Input image. downscale : float, optional Downscale factor. sigma : float, optional Sigma for Gaussian filter. Default is `2 * downscale / 6.0` which corresponds to a filter mask twice the size of the scale factor that covers more than 99% of the Gaussian distribution. order : int, optional Order of splines used in interpolation of downsampling. See `skimage.transform.warp` for detail. mode : {'reflect', 'constant', 'edge', 'symmetric', 'wrap'}, optional The mode parameter determines how the array borders are handled, where cval is the value when mode is equal to 'constant'. cval : float, optional Value to fill past edges of input if mode is 'constant'. multichannel : bool, optional Whether the last axis of the image is to be interpreted as multiple channels or another spatial dimension. This argument is deprecated: specify `channel_axis` instead. preserve_range : bool, optional Whether to keep the original range of values. Otherwise, the input image is converted according to the conventions of `img_as_float`. Also see https://scikit-image.org/docs/dev/user_guide/data_types.html channel_axis : int or None, optional If None, the image is assumed to be a grayscale (single channel) image. Otherwise, this parameter indicates which axis of the array corresponds to channels. .. versionadded:: 0.19 ``channel_axis`` was added in 0.19. Returns ------- out : array Smoothed and downsampled float image. References ---------- .. [1] http://persci.mit.edu/pub_pdfs/pyramid83.pdf Nc32|]\}}|krt|tn|VqdSNmathceilfloat.0axdr downscalerr W  z!pyramid_reduce..c3"|] }t|tVqdSrrr#r%r'rrr(\s r@Forderr r anti_aliasing)rrrtuple enumerateshaperr) rr'rr/r r multichannelpreserve_ranger out_shaperoutrr&rpyramid_reduce s2     r8c stt||}dur"|jtfddt|jD} n tfdd|jD} |dur8dd}t|| |||dd} t| |||} | S) a Upsample and then smooth image. Parameters ---------- image : ndarray Input image. upscale : float, optional Upscale factor. sigma : float, optional Sigma for Gaussian filter. Default is `2 * upscale / 6.0` which corresponds to a filter mask twice the size of the scale factor that covers more than 99% of the Gaussian distribution. order : int, optional Order of splines used in interpolation of upsampling. See `skimage.transform.warp` for detail. mode : {'reflect', 'constant', 'edge', 'symmetric', 'wrap'}, optional The mode parameter determines how the array borders are handled, where cval is the value when mode is equal to 'constant'. cval : float, optional Value to fill past edges of input if mode is 'constant'. multichannel : bool, optional Whether the last axis of the image is to be interpreted as multiple channels or another spatial dimension. This argument is deprecated: specify `channel_axis` instead. preserve_range : bool, optional Whether to keep the original range of values. Otherwise, the input image is converted according to the conventions of `img_as_float`. Also see https://scikit-image.org/docs/dev/user_guide/data_types.html channel_axis : int or None, optional If None, the image is assumed to be a grayscale (single channel) image. Otherwise, this parameter indicates which axis of the array corresponds to channels. .. versionadded:: 0.19 ``channel_axis`` was added in 0.19. Returns ------- out : array Upsampled and smoothed float image. References ---------- .. [1] http://persci.mit.edu/pub_pdfs/pyramid83.pdf Nc3s.|]\}}|krt|n|VqdSrrr r"r upscalerrr(s  z!pyramid_expand..c3s|] }t|VqdSrr9r+)r;rrr(srr-Fr.)rrrr1r2r3rr) rr;rr/r r r4r5r r6resizedr7rr:rpyramid_expandis2    r=c  cs~t|t||}d} |j} |} |V| |kr=| d7} t| |||||| d} | }| } | j} | |kr4dS| V| |ksdSdS)an Yield images of the Gaussian pyramid formed by the input image. Recursively applies the `pyramid_reduce` function to the image, and yields the downscaled images. Note that the first image of the pyramid will be the original, unscaled image. The total number of images is `max_layer + 1`. In case all layers are computed, the last image is either a one-pixel image or the image where the reduction does not change its shape. Parameters ---------- image : ndarray Input image. max_layer : int, optional Number of layers for the pyramid. 0th layer is the original image. Default is -1 which builds all possible layers. downscale : float, optional Downscale factor. sigma : float, optional Sigma for Gaussian filter. Default is `2 * downscale / 6.0` which corresponds to a filter mask twice the size of the scale factor that covers more than 99% of the Gaussian distribution. order : int, optional Order of splines used in interpolation of downsampling. See `skimage.transform.warp` for detail. mode : {'reflect', 'constant', 'edge', 'symmetric', 'wrap'}, optional The mode parameter determines how the array borders are handled, where cval is the value when mode is equal to 'constant'. cval : float, optional Value to fill past edges of input if mode is 'constant'. multichannel : bool, optional Whether the last axis of the image is to be interpreted as multiple channels or another spatial dimension. This argument is deprecated: specify `channel_axis` instead. preserve_range : bool, optional Whether to keep the original range of values. Otherwise, the input image is converted according to the conventions of `img_as_float`. Also see https://scikit-image.org/docs/dev/user_guide/data_types.html channel_axis : int or None, optional If None, the image is assumed to be a grayscale (single channel) image. Otherwise, this parameter indicates which axis of the array corresponds to channels. .. versionadded:: 0.19 ``channel_axis`` was added in 0.19. Returns ------- pyramid : generator Generator yielding pyramid layers as float images. References ---------- .. [1] http://persci.mit.edu/pub_pdfs/pyramid83.pdf rrrN)rrr3r8)r max_layerr'rr/r r r4r5r layer current_shapeprev_layer_image layer_image prev_shaperrrpyramid_gaussians&=  rFc  #stt||}|durdd}|j} t||||} || Vdur;|jt| } | t| } n| } |dkrLt t t | }t |D]:} duretfddt | D}n tfdd| D}t| ||||dd }t||||} |j} || VqPdS) aI Yield images of the laplacian pyramid formed by the input image. Each layer contains the difference between the downsampled and the downsampled, smoothed image:: layer = resize(prev_layer) - smooth(resize(prev_layer)) Note that the first image of the pyramid will be the difference between the original, unscaled image and its smoothed version. The total number of images is `max_layer + 1`. In case all layers are computed, the last image is either a one-pixel image or the image where the reduction does not change its shape. Parameters ---------- image : ndarray Input image. max_layer : int, optional Number of layers for the pyramid. 0th layer is the original image. Default is -1 which builds all possible layers. downscale : float, optional Downscale factor. sigma : float, optional Sigma for Gaussian filter. Default is `2 * downscale / 6.0` which corresponds to a filter mask twice the size of the scale factor that covers more than 99% of the Gaussian distribution. order : int, optional Order of splines used in interpolation of downsampling. See `skimage.transform.warp` for detail. mode : {'reflect', 'constant', 'edge', 'symmetric', 'wrap'}, optional The mode parameter determines how the array borders are handled, where cval is the value when mode is equal to 'constant'. cval : float, optional Value to fill past edges of input if mode is 'constant'. multichannel : bool, optional Whether the last axis of the image is to be interpreted as multiple channels or another spatial dimension. This argument is deprecated: specify `channel_axis` instead. preserve_range : bool, optional Whether to keep the original range of values. Otherwise, the input image is converted according to the conventions of `img_as_float`. Also see https://scikit-image.org/docs/dev/user_guide/data_types.html channel_axis : int or None, optional If None, the image is assumed to be a grayscale (single channel) image. Otherwise, this parameter indicates which axis of the array corresponds to channels. .. versionadded:: 0.19 ``channel_axis`` was added in 0.19. Returns ------- pyramid : generator Generator yielding pyramid layers as float images. References ---------- .. [1] http://persci.mit.edu/pub_pdfs/pyramid83.pdf .. [2] http://sepwww.stanford.edu/data/media/public/sep/morgan/texturematch/paper_html/node3.html Nrr-r?c3rrrr"r&rrr(kr)z$pyramid_laplacian..c3r*rrr+r,rrr(psFr.)rrr3rrlistpopr1rr logmaxranger2r)rr@r'rr/r r r4r5r rBsmoothed_imageshape_without_channelsrAr6 resized_imagerr&rpyramid_laplacian sBA          rO)rNrrrFF)r?rNrrrFF)rnumpyr _sharedr_shared.filtersr _shared.utilsr transformrrrdeprecate_multichannel_kwargr8r=rFrOrrrrsH      H G Z