U
    ?h                     @   s,   d dl ZddlmZ dejd dfddZdS )    N   )view_as_blocksc           	      C   s   t |r|f| j }nt|| jkr.td|dkr:i }g }tt|D ]\}|| dk rbtd| j| ||  dkr|| | j| ||   }nd}|d|f qJt j| |d|d} t	| |}||fdt
t| j|ji|S )	a  Downsample image by applying function `func` to local blocks.

    This function is useful for max and mean pooling, for example.

    Parameters
    ----------
    image : ndarray
        N-dimensional input image.
    block_size : array_like or int
        Array containing down-sampling integer factor along each axis.
        Default block_size is 2.
    func : callable
        Function object which is used to calculate the return value for each
        local block. This function must implement an ``axis`` parameter.
        Primary functions are ``numpy.sum``, ``numpy.min``, ``numpy.max``,
        ``numpy.mean`` and ``numpy.median``.  See also `func_kwargs`.
    cval : float
        Constant padding value if image is not perfectly divisible by the
        block size.
    func_kwargs : dict
        Keyword arguments passed to `func`. Notably useful for passing dtype
        argument to ``np.mean``. Takes dictionary of inputs, e.g.:
        ``func_kwargs={'dtype': np.float16})``.

    Returns
    -------
    image : ndarray
        Down-sampled image with same number of dimensions as input image.

    Examples
    --------
    >>> from skimage.measure import block_reduce
    >>> image = np.arange(3*3*4).reshape(3, 3, 4)
    >>> image # doctest: +NORMALIZE_WHITESPACE
    array([[[ 0,  1,  2,  3],
            [ 4,  5,  6,  7],
            [ 8,  9, 10, 11]],
           [[12, 13, 14, 15],
            [16, 17, 18, 19],
            [20, 21, 22, 23]],
           [[24, 25, 26, 27],
            [28, 29, 30, 31],
            [32, 33, 34, 35]]])
    >>> block_reduce(image, block_size=(3, 3, 1), func=np.mean)
    array([[[16., 17., 18., 19.]]])
    >>> image_max1 = block_reduce(image, block_size=(1, 3, 4), func=np.max)
    >>> image_max1 # doctest: +NORMALIZE_WHITESPACE
    array([[[11]],
           [[23]],
           [[35]]])
    >>> image_max2 = block_reduce(image, block_size=(3, 1, 4), func=np.max)
    >>> image_max2 # doctest: +NORMALIZE_WHITESPACE
    array([[[27],
            [31],
            [35]]])
    zF`block_size` must be a scalar or have the same length as `image.shape`N   zYDown-sampling factors must be >= 1. Use `skimage.transform.resize` to up-sample an image.r   Zconstant)	pad_widthmodeZconstant_valuesZaxis)npZisscalarndimlen
ValueErrorrangeshapeappendpadr   tuple)	image
block_sizefunccvalZfunc_kwargsr   iZafter_widthZblocked r   G/var/www/html/venv/lib/python3.8/site-packages/skimage/measure/block.pyblock_reduce   s*    :


r   )numpyr   utilr   sumr   r   r   r   r   <module>   s   