U ?h3@sddlZddlZddlmZddlmZddlmZddZ dd Z ddd ddZ ddd ddZ ddd ddZ ddd ddZdS)N)gaussian)convert_to_float)resizecCsJt|}|dk r.||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 smoothedrL/var/www/html/venv/lib/python3.8/site-packages/skimage/transform/pyramids.py_smooth s   rcCs|dkrtddS)Nrz#scale factor must be greater than 1) ValueError)factorrrr _check_factorsrrreflectFr c stt||}dk rD|jtfddt|jD}ntfdd|jD}|dkrpdd}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 Nc3s0|](\}}|kr$t|tn|VqdSNmathceilfloat.0Zaxdr downscalerr Qsz!pyramid_reduce..c3s |]}t|tVqdSrrrrr!rrr"Vsr@Forderrr Z anti_aliasing)rrr tuple enumerateshaperr) r r!rr'rr preserve_ranger out_shaperoutrr rpyramid_reduces-     r.c stt||}dk rD|jtfddt|jD}ntfdd|jD}|dkrpdd}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,|]$\}}|kr t|n|VqdSrrrrr upscalerrr"sz!pyramid_expand..c3s|]}t|VqdSrr/r#)r1rrr"srr%Fr&)rrr r(r)r*rr) r r1rr'rr r+r r,Zresizedr-rr0rpyramid_expandcs"-    r2c cspt|t||}d} |j} |} |V| |krl| d7} t| ||||||d} | } | } | j} | | krdql| Vq&dS)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*r.)r max_layerr!rr'rr r+r layer current_shapeZprev_layer_imageZ layer_imageZ prev_shaperrrpyramid_gaussians&8  r7c #stt||}|dkr&dd}|j} t||||} || Vdk rt|jt| } | t| } n| } |dkrt t t | }t |D]t} dk rtfddt | D} ntfdd| D} t| | |||dd }t||||} |j} || VqdS) 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%r3c3s0|](\}}|kr$t|tn|VqdSrrrr rrr"Vsz$pyramid_laplacian..c3s |]}t|tVqdSrrr#r$rrr"[sFr&)rrr*rr listpopr(rrlogmaxranger)r)r r4r!rr'rr r+r r6Zsmoothed_imageZshape_without_channelsr5r,Z resized_imagerr rpyramid_laplaciansB<          r=)rNrrrF)rNrrrF)r3rNrrrF)r3rNrrrF)rnumpyr Z_shared.filtersrZ _shared.utilsrZ transformrrrr.r2r7r=rrrrsD    D C V