U ?h @s:ddlZddlmZddZejddddd d ZdS) N)utilsc Cs|jjdkrH|d}t|}t|d}t|d}||}n4tj|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)Zreturn_inverse return_counts)r) dtypekindZreshapenpZbincountZnonzerouniqueZcumsumsizeinterpshape) sourcetemplateZ src_lookupZ src_countsZ tmpl_countsZ tmpl_valuesZ src_valuesZ src_quantilesZtmpl_quantilesZinterp_a_valuesrU/var/www/html/venv/lib/python3.8/site-packages/skimage/exposure/histogram_matching.py_match_cumulative_cdfs"        r)r)Zchannel_arg_positions) channel_axiscCs|j|jkrtd|dk r|jd|jdkr8tdtj|j|jd}t|jdD]*}t|d|f|d|f}||d|f<qXn t||}|jjdkrt |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 ValueErrorr r emptyrrangerrrZ_supported_float_typeZastype)image referencerZmatchedZchannelZmatched_channelZ out_dtyperrrmatch_histograms"s "      r)numpyr Z_sharedrrZchannel_as_last_axisrrrrrs