o eB5@sddlZddlZddlmZddlmZddlmZddZ dd Z ddd ddZ ddd ddZ ddd ddZ ddd ddZdS)N)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)npZ empty_likendimr)imagesigmarr r smoothedrJD:\Projects\ConvertPro\env\Lib\site-packages\skimage/transform/pyramids.py_smooth s   rcCs|dkrtddS)Nrz#scale factor must be greater than 1) ValueError)factorrrr _check_factorsrrreflectFr c stt||}dur"|jtfddt|jD}n tfdd|jD}|dur8dd}t||||} t| ||||dd} | S) a6Smooth 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'. 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 Q  z!pyramid_reduce..c3"|] }t|tVqdSrrrr!r#rrr$Vs r@Forderrr Z anti_aliasing)rrr tuple enumerateshaperr) r r#rr+rr preserve_ranger out_shaperoutrr"rpyramid_reduces-     r2c 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'. 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|VqdSrrrrr upscalerrr$s  z!pyramid_expand..c3s|] }t|VqdSrr3r')r5rrr$srr)Fr*)rrr r,r-r.rr) r r5rr+rr r/r r0Zresizedr1rr4rpyramid_expandcs-    r6c cs~t|t||}d} |j} |} |V| |kr=| d7} t| ||||||d} | } | } | j} | | kr4dS| V| |ksdSdS)aYield 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'. 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)rrr.r2)r max_layerr#rr+rr r/r layer current_shapeZprev_layer_imageZ layer_imageZ prev_shaperrrpyramid_gaussians&8  r;c #stt||}|durdd}|j} t||||} || Vdur;|jt| } | t| } n| } |dkrLt t t | }t |D]:} duretfddt | D} n tfdd| D} t| | |||dd }t||||} |j} || VqPdS) ag 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'. 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)r7c3rrrrr"rrr$Vr%z$pyramid_laplacian..c3r&rrr'r(rrr$[sFr*)rrr.rr listpopr,rrlogmaxranger-r)r r8r#rr+rr r/r r:Zsmoothed_imageZshape_without_channelsr9r0Z resized_imagerr"rpyramid_laplaciansB<          rA)rNrrrF)r7rNrrrF)rnumpyr Z_shared.filtersrZ _shared.utilsrZ transformrrrr2r6r;rArrrrs>    D C V