o h @s:ddlZddlmZddZejddddd d ZdS) N)utilsc Cs|jjdkr$|d}t|}t|d}t|d}||}ntj|dddd\}}}tj|ddd\}}t||j}t||j}t |||} | ||j S)z Return modified source array so that the cumulative density function of its values matches the cumulative density function of the template. urT)return_inverse return_counts)r) dtypekindreshapenpbincountnonzerouniquecumsumsizeinterpshape) sourcetemplate src_lookup src_counts tmpl_counts tmpl_values src_values src_quantilestmpl_quantilesinterp_a_valuesr^/var/www/html/scripts/venv/lib/python3.10/site-packages/skimage/exposure/histogram_matching.py_match_cumulative_cdfs"       r)r)channel_arg_positions) channel_axiscCs|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) awAdjust 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. 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.NrzENumber of channels in the input image and reference image must match!)r.fF)copy) ndim ValueErrorrr emptyrrangerr r_supported_float_typeastype)image referencer"matchedchannelmatched_channel out_dtyperrrmatch_histograms"s" "     r1)numpyr _sharedrrchannel_as_last_axisr1rrrrs