o :ήc @sDddlZddlmZddZejddeddd d d ZdS) N)utilsc Csntj|ddd\}}}tj|dd\}}t||j}t||j}t|||} | ||jS)z Return modified source array so that the cumulative density function of its values matches the cumulative density function of the template. T)return_inverse return_counts)r)npuniqueravelcumsumsizeinterpreshapeshape) sourcetemplate src_valuessrc_unique_indices src_counts tmpl_values tmpl_counts src_quantilestmpl_quantilesinterp_a_valuesrJ/tmp/pip-target-vg8gfxp4/lib/python/skimage/exposure/histogram_matching.py_match_cumulative_cdfs  r)r)channel_arg_positionsF) channel_axis multichannelcCs|j|jkr td|durC|jd|jdkrtdtj|j|jd}t|jdD]}t|d|f|d|f}||d|f<q,nt||}|jjdkr[t |j}|j |dd }|S) aAdjust an image so that its cumulative histogram matches that of another. The adjustment is applied separately for each channel. Parameters ---------- image : ndarray Input image. Can be gray-scale or in color. reference : ndarray Image to match histogram of. Must have the same number of channels as image. 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. multichannel : bool, optional Apply the matching separately for each channel. This argument is deprecated: specify `channel_axis` instead. Returns ------- matched : ndarray Transformed input image. Raises ------ ValueError Thrown when the number of channels in the input image and the reference differ. References ---------- .. [1] http://paulbourke.net/miscellaneous/equalisation/ z:Image and reference must have the same number of channels.NzENumber of channels in the input image and reference image must match!)dtype.fF)copy) ndim ValueErrorr remptyr rangerkindr_supported_float_typeastype)image referencerrmatchedchannelmatched_channel out_dtyperrrmatch_histogramss" '     r0)numpyr_sharedrrchannel_as_last_axisdeprecate_multichannel_kwargr0rrrrs