a "Df@sddlZddlZdddZdS)NFc Csp|jdkrd}n|jdks"Jd|jd}|durBtj|dd}t|t}|j|fksbJ|durxtj|dd}t|t}|j|fksJt| st| rt dt |j j}||k}|||8<|||7<tj|jtd }|jdkrt|}nt|j dd}d |||||||<tj||<|rh|||fS|SdS) axNormalize image. Parameters ---------- src: numpy.ndarray, (H, W) or (H, W, C), float Input image. min_value: float Minimum value. max_value: float Maximum value. return_minmax: bool Flag to return min_value and max_value. Returns ------- dst: numpy.ndarray, float Normalized image in [0, 1]. zsrc ndim must be 2 or 3N)rr)Zaxisz"some of min or max values are inf.)dtypeg?)ndimshapenpZnanminZ atleast_1dZastypefloatZnanmaxisinfanywarningswarnZfinforepszerosisnannan) srcZ min_valueZ max_valueZ return_minmaxDrZissamedstrr]/nfs/NAS7/SABIOD/METHODE/ermites/ermites_venv/lib/python3.9/site-packages/imgviz/normalize.py normalizes6        r)NNF)r numpyrrrrrrs