a d @sddlZddlZddlZddlmZmZmZmZm Z ddl m Z ddl m Z ddl m Z ddlmZddlmZddlmZdd lmZdd lmZejd ejejgd d ZddZddZddZddZ ddZ!ddZ"ddZ#ddZ$ejd ej%ejej&gddZ'ejd ej%ejej&gd d!Z(d"d#Z)ejd$d d%gejd ej*ej+ej&gejd&gd'd(d)Z,ej-.de /Z0e 1e0Z2e 3e2d*d+Z2d,d-Z4d.d/Z5ejd ej%ejej&gd0d1Z6d2d3Z7d4d5Z8ejd6d%dd7gfd dd8gfd9d:d;gfgdd?gfd d@d7gfdAdBdCgfgdDdEZ:dFdGZ;dHZdKZ?dLdMZ@dNdOZAejd ej%ejej&gdPdQZBdRdSZCejd ejejDej%ejej&gdTdUZEdVdWZFdXdYZGdZd[ZHd\d]ZId^d_ZJd`daZKdbdcZLejdddedfgdgdhZMejdidjej*fdkejNfdlejNfdmejNfdneOfgdodpZPdqdrZQdsdtZRdudvZSejd ej%ejej&gdwdxZTdydzZUd{d|ZVd}d~ZWddZXddZYddZZddZ[ddZ\ddZ]ddZ^ddZ_ejd ej%ejej&gddZ`ddZaddZbddZcddZdejd ej%ejej&gddZeddZfddZgejd ej%ejej&gddZhddZiddZjddZkddZlddZmddZnejde joe jpe jqgddZrddZsddZtdS)N)assert_allcloseassert_almost_equalassert_array_almost_equalassert_array_equal assert_equal)data)exposure)util)rgb2gray)intensity_range) dtype_range)expected_warnings)_supported_float_typedtypecCsVtjddg|d}tjtdd"tj|dd\}}Wdn1sH0YdS)Ndrz+Incorrect value for `source_range` argumentmatchZfoobar source_range)nparraypytestraises ValueErrorr histogramrim frequencies bin_centersr!m/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/skimage/exposure/tests/test_exposure.pytest_wrong_source_ranges r#cCsjtjddgtjd}t|\}}t|tdd|ddksDJ|ddksTJt|ddddS)Nrrrerrrint8rrrarangerrr r!r!r"test_negative_overflow s r*cCsjtjddgtjd}t|\}}t|tdd|ddksDJ|ddksTJt|ddddS)Nirrrr%r&r)r!r!r"test_all_negative_image)s r+cCsTtjddgtjd}t|\}}tt|t|t|ddt|dddS)N rrrr)rrr'rrrlenr)r!r!r"test_int_range_image2s r.cCsxtjddgddggtjd}tj|dd\}}t|jdD]}tt||t|qnJdS)z=Slope of cdf which should equal 1 for an equalized histogram.rr%rJ皙?N)rrAr-Zpolyfit)r`Znorm_intensityZslopeZ interceptr!r!r"r]sr]ztest_input,expectedr%r@)r,r,rlcCs.tjddgtjd}t||d}t||dS)Nrr%r range_values)rrr8r rZ test_inputrQrSoutr!r!r"test_intensity_range_uint8s rq皙?皙?r)333333?皙?rtrucCs.tjddgtjd}t||d}t||dS)Nrrrsrrmrrfloat64r rror!r!r"test_intensity_range_floats rxcCs0tjddgtjd}t|ddd}t|ddS)NrrrsrrT)rnZ clip_negative)rr%rvrSrpr!r!r""test_intensity_range_clipped_floatsrziii?icCs@tjgdtjd}t|}|jtjks.Jt|gddS)N3r0rrr@)rrr8rrescale_intensityrrryr!r!r"test_rescale_stretch s rcCs*tgd}t|}t|gddS)NgI@gY@g c@)rrGr%rrrrrryr!r!r"test_rescale_shrinks rcCsFtjgd|d}tj|dd}t|gddd|j|jksBJdS)Nrrrr@in_range)rsrug333333?)decimal)rrrrrrrrSrpr!r!r"test_rescale_in_rangesrcCs.tgd}tj|dd}t|gddS)Nr)rr0r)rGr%r%rryr!r!r"test_rescale_in_range_clip srcCsFtjgd|d}tj|dd}|jt|jks4Jt|gddS)a?Check that output range is correct. .. versionchanged:: 0.17 This function used to return dtype matching the input dtype. It now matches the output. .. versionchanged:: 0.19 float16 and float32 inputs now result in float32 output. Formerly they would give float64 outputs. )rr,rrr out_range)rgO@rN)rrrrrrrrr!r!r"test_rescale_out_range&s rcCs<tjdttdgtjd}tj|dd}t|dttgdS)Nrrruint10r)rr uint10_maxuint16rrr uint16_maxryr!r!r"test_rescale_named_in_range9srcCs4tjdtgtjd}tj|dd}t|dtgdS)Nrrrr)rrrrrrrrryr!r!r"test_rescale_named_out_range?srcCs4tjdtgtjd}tj|dd}t|dtgdS)Nrruint12r)rrrrrrr uint12_maxryr!r!r"test_rescale_uint12_limitsEsrcCs4tjdtgtjd}tj|dd}t|dtgdS)NrrZuint14r)rrrrrrr uint14_maxryr!r!r"test_rescale_uint14_limitsKsrcCs<tjdtjd}t|}t|s.Jt||dS)NrOrOr)rrPr8rrisnanrirryr!r!r"test_rescale_all_zerosQs rcCs4tjddgtjd}tj|dd}t|ddgdS)Nrrrr)rrrrrrryr!r!r"test_rescale_constantXsrcCs6td}t|}t|s(Jt||dS)Nr)ronesrrrrirryr!r!r"test_rescale_same_values^s  rzin_range,out_range)rSr)rrScCsztjdtddd}tj|d<d}d}|dkr:||g}n|g}t|t|||Wdn1sl0YdS) N rrIr)r%r%zWOne or more intensity levels are NaN\. Rescaling will broadcast NaN to the full image\.z/Passing `np.nan` to mean no clipping in np.cliprS)rr(floatreshapenanr rr)rrrSmsgZnumpy_warning_1_17_plusZexp_warnr!r!r"test_rescale_nan_warninges    rzout_range, out_dtyper8rrrrcCs4tjgdtjd}tj||d}|j|ks0JdS)Nr:rrrr)rrr'rrr)rZ out_dtyperS output_imager!r!r"test_rescale_output_dtypes rcCsFtjgdtjd}tj|tjd}t|gd|jtjksBJdS)Nrrrrr;r@)rrr'rrr8rrrSrr!r!r"test_rescale_no_overflowsrcCsBtjgdtjd}tj|dd}t|gd|jtks>JdS)Nrrrrr)rrr'rrrrrrr!r!r"test_rescale_float_outputsrcCsPtjgdtjd}tttj|dd}Wdn1sB0YdS)NrrZflatr)rrr'rrrrr)rS_r!r!r"*test_rescale_raises_on_incorrect_out_ranges rcCsttj|dd}t|}t|||f}tj |dddd}|j |j ksRJ|j t |ksdJ|tj krrdnd}tt||d |tt||d dd S) z!Test a grayscale float image Frc)9r|{Gz?r;) kernel_size clip_limitrFrIrOg)\Y@ghs?N)r rer astronautrfr rdstackrequalize_adapthistr3rrfloat16rpeak_snrnorm_brightness_err)rr^adaptedZ snr_decimalr!r!r"test_adapthist_grayscalesrcCstt}tjdd8}tdt|\}}t |dksDJWdn1sX0Ytj |dd}| dksJ| dksJ|j |j ksJt|}tt||d d tt||d d dS) z#Test an RGB color uint16 image T)recordalwaysrNrr?x&Y[@r%{Gz?rO)r Z img_as_uintrrwarningscatch_warnings simplefilterrrr-rrKrLr3rrrr)r^whistr r full_scaler!r!r"test_adapthist_colors . rcCstt}tj|jd|jdftd}t||f}t |}|j|jksTJ|ddddddf}t |}|j|jksJt t ||ddt t||dddS) zTest an RGBA color image rr%rNrIrrOga+e?)r rerrrrr3rrrrrrrr)r^alpharrr!r!r"test_adapthist_alphas  rcCstt}t|}d}t|dd|dd|f}t|g|jd|}tj |ddd}tj |ddd}|j|jksJ|j|jksJt t |||jdddksJd S) z Test for n-dimensional consistency with float images Note: Currently if img.ndim == 3, img.shape[2] > 4 must hold for the image not to be interpreted as a color image by @adapt_rgb rrr/g?)rrrOrN) r rerrr rrr3rrmeanabs)r^aZimg2dZimg3dZ adapted2dZ adapted3dr!r!r"test_adapthist_grayscale_Nds( rcCstd}|d7}|tj}t|d}t|t|ksBJtd}|d7}|tj}t|d}t|t|ksJdS)z(Test constant image, float and uint rrOrIrrN) rrPrfrrrrKrLrw)r^rr!r!r"test_adapthist_constants      rcCsttt}|d}d||jdd|jddf<d}tdddD]h}tj||d d }t |d d |f|d d |fd ksJt ||d d f||d d fd ksJJqJd S) zTest border processing gY@rrrOr%rr|GrGrNrr) r r rerrr3r2rrr)r^Z border_indexrrr!r!r"test_adapthist_borderss  rcCsbt}t|}tj|dd}tj|dd}t||tj|dd}tj|dd}t||dS)Nrrr%)rZmoonr rerrr)Zimg_uZimg_fZ img_clahe0Z img_clahe1r!r!r"test_adapthist_clip_limits  rcCsv|jdkr$t|t|}}t|}t|}d|jt||}t |j j \}}dt ||S)zPeak signal to noise ratio of two images Parameters ---------- img1 : array-like img2 : array-like Returns ------- peak_snr : float Peak signal to noise ratio rIrrl) ndimr rdr resizerZsquaresumr rtypelog)img1img2ZmserZmax_r!r!r"r,s   rcCsJ|jdkrt|t|}}t||}|t|jjd}|S)zNormalized Absolute Mean Brightness Error between two images Parameters ---------- img1 : array-like img2 : array-like Returns ------- norm_brightness_error : float Normalized absolute mean brightness error rIr%)rr rrrr rr)rrZambeZnber!r!r"rBs rcCsLtjdtd}tjtddt|dWdn1s>0YdS)Nrrz Incorrect value of `kernel_size`r)rIrIrI)rrrrrrrrr^r!r!r"$test_adapthist_incorrect_kernel_sizeVsrcCs.tddg}t|d}|j|jks*JdS)"Check that the shape is maintainedr%g?N)rrr adjust_gammar3)r^resultr!r!r"test_adjust_gamma_1x1_shape_s rcCs2tjddtjdd}t|d}t||dS)z4Same image should be returned for gamma equal to onerr6rrr%N)rr(r8rrrr)rSrr!r!r"test_adjust_gamma_onefs rcCsTtjdddj|dd}t|d}|jj}t|t |d|j|jksPJdS)z6White image should be returned for gamma equal to zerorr@rFrcr%N) rrandomuniformrfrrrrrr )rrSrr!r!r"test_adjust_gamma_zeroms  rcCsLtjddtjdd}tjgdtjdd}t|d}t||dS)z\Verifying the output with expected results for gamma correction with gamma equal to halfrr6rr(rr $'*-0257:<>@BDFrIKMNPQSTVWYrD\]^`abrr$r0gijklmopqrstuvwyz{|}~rr;rr)r<r2r8r}r<r@rDrGrJrMrPrSrUrWrZr\r^r`rbrdrfrgrirkrlrnrprqrsrtrvrwryrzr|r}r~rrrr=r=r>r>rrrrrrrrrrrrrrrrrrrrrrrr@rGNrr(r8rrrrrrSrQrr!r!r"test_adjust_gamma_less_onews  rcCsLtjddtjdd}tjgdtjdd}t|d}t||dS)z[Verifying the output with expected results for gamma correction with gamma equal to tworr6rr(rrrrrrrrrrrrr%r%r%r%r%r%r%r%rOrOrOrOrOrOrIrIrIrIrrrrr/r/r/r/rrrrrrr rrr,r, rrrr rrrrrrrrrrlrlrrrrrrrrrr!"#rr%&rr()rr+,r./rr1rr|4r6r8rrr;r=r?rArCrErrrJrLrrOrrRrUrrXr[r r r _r rcrr0rhrrrrrrrrrrrxrr r!r#rr;rr%r'r(r*r+r,r-r/r0r2r3r5r6r8r9r:r;r=r>r@rBrCrErFrHrJrKrMrOrPrRrTrUrWrYrZr\r^r`rarcrergrhrjrlrnrorqrsrurwryrzr|r~rrr=rrrrrrrrrrrrrr@rONrrr!r!r"test_adjust_gamma_greater_ones  rcCsRtdddtjd}ttt|dWdn1sD0YdS)Nrr@rrr) rr(r8rrrrrrrSr!r!r"test_adjust_gamma_negatives rcCs6dtjdtjd}ttj|ddddks2JdS)Nr@rrr%rk)gammaZgain)rrr8rirrrr!r!r"test_adjust_gamma_u8_overflowsrcCs@tjddg|d}t|d}|j|jks.J|j|ksrrrrrr%TNrrr!r!r"test_adjust_inv_logs rcCsBtjddg|d}t|dd}|j|jks0J|j|ks>JdS)rr%rr/N)rrradjust_sigmoidr3rrr!r!r"test_adjust_sigmoid_1x1_shapesrc Csttdddtjd}tjgdgdgdgdgd gd gd gd gtjd }t|dd}t||dS)zlVerifying the output with expected results for sigmoid correction with cutoff equal to one and gain of 5rr@rr)r%r%r%rOrOrOrOrO)rIrIrIrrrr/r/)r/rrrrrrr,)r,rrrrrrr)rrlrrrrrr)rrrrrrrr)rrrrrrrr)rr rrrrrr rr%r/Nrr(r8rrrrrrr!r!r"test_adjust_sigmoid_cutoff_one s rc Csttdddtjd}tjgdgdgdgdgd gd gd gd gtjd }t|dd }t||d S)znVerifying the output with expected results for sigmoid correction with cutoff equal to zero and gain of 10rr@rr)rr+r4r<rFrOrWr_)rfrmrsrxr}rr>r)rrrrrrrrrrrrrrrr)rrrrrrrr)rrrrrrrrrr,Nrrr!r!r"test_adjust_sigmoid_cutoff_zeros rc Csttdddtjd}tjgdgdgdgdgd gd gd gd gtjd }t|dd}t||dS)znVerifying the output with expected results for sigmoid correction with cutoff equal to half and gain of 10rr@rr)r%r%rOrOrIrIrr/)r/rrrr,rrr)rrrrrrrr)rrrrrrrr)r;r<r5r>rGrPrXr`)rgrmrsryr}rr>r)rrrrrrrrrrrGr,Nrrr!r!r"test_adjust_sigmoid_cutoff_half-s rc Csvtdddtjd}tjgdgdgdgdgd gd gd gd gtjd }t|ddd}t||dS)zvVerifying the output with expected results for inverse sigmoid correction with cutoff equal to half and gain of 10rr@rr)rrrrrrrr)rrrrrrrr)rrr>rr|rwrrrl)rer^rVrNrEr;r3r*)r#rrr rrrr)rrrrrrrr)rrrrr,rrr)r/rrrIrIrOrOr%rrGr,TNrrr!r!r"#test_adjust_inv_sigmoid_cutoff_half?s rcCstddd}t|sJd|d<t|s2Jtj|ddrDJ|dtj}t|sbJtj|ddrtJ|tjd}t|sJtj|ddrJdS) Nrg{Gz?rr%r)Zupper_percentiler@r6)rrAris_low_contrastrfr8rrr!r!r"test_is_low_contrastQsrcCs:tjdtd}t|sJd|dd<t|r6JdS)Nrrr%r/)rrPboolrrrr!r!r"test_is_low_contrast_booleanas r exposure_funccCsRtddddtj}tt||Wdn1sD0YdS)Nrrrr)rr(rrfrwrrr)rrSr!r!r"test_negative_inputls rcCstjdddddlm}|jtddgddggdd}t|\}}gd }gd }t||shJt||sxJt ||j sJdS) Ndask$dask python library is not installedreasonrr%rO)r%rO)chunks)rr%rO)r%rOr%) r importorskip dask.arrayrZ from_arrayrrrZallclose isinstanceArray)daZ dask_arrayZ output_histZ output_binsZ expected_binsZ expected_histr!r!r"test_dask_histogramxs  rcCsjtjdddddlm}|jgdtjd}t|}|jtjksHJt |gdt ||j sfJdS)Nrrrrr{rr~) rrrrrr8rrrrrr)rrSrpr!r!r"test_dask_rescales  r)urnumpyrrZ numpy.testingrrrrrZskimagerrr Z skimage.colorr Zskimage.exposure.exposurer Zskimage.util.dtyper Zskimage._shared._warningsr Zskimage._shared.utilsrmarkZ parametrizer'float32r#r*r+r.r5r9r?rBrCrrwrHrMrRr8int16rWrseedZcamerarYrer[rrZrbrgrjr]rqrxrzrrrrrrrrint32rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrr!r!r!r"s                                      '#