1-Phz ZddlZddlmZdZejddddZdS) N)utilsc|jjdkrs|d}tj|}tj|d}tj|d}||}nZtj|ddd\}}}tj|dd\}}tj||jz }tj||jz }tj |||} | ||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_valuess c/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/skimage/exposure/histogram_matching.py_match_cumulative_cdfr s) |C^^B'' [,, k("2"22"6"677 j--a0 !+. -/Y NN2  t4. . . * J $&9X-=-=b-A-AQU#V#V#V [Ij))FK7MY{++hm;Ni ~{KKO : & . .v| < <<)r)channel_arg_positions) channel_axisc|j|jkrtd||jd|jdkrtdtj|j|j}t |jdD])}t|d|f|d|f}||d|f<*nt||}|jjdkr0tj |j}| |d }|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 emptyr ranger r r_supported_float_typeastype)image referencer$matchedchannelmatched_channel out_dtypes rmatch_histogramsr4!s"D zY^## K    ;r?iob1 1 1.  (5;ek:::U[_-- 4 4G3c7l#YsG|%<O%4GCL ! !  4(y99}S  / << ...77 Nr!)numpyr _sharedrr channel_as_last_axisr4r!rr9su===6&9997;;;;;:9;;;r!