o
    ,h                     @   sR   d Z ddlZddlmZ ddlZdd Zddd	Zdd
dZdddZ	dd Z
dS )z6  
Copyright (c) 2019-present NAVER Corp.
MIT License
    N)ioc                 C   sr   t | }|jd dkr|d }t|jdkrt|tj}|jd dkr2|d d d d d df }t|}|S )Nr            )	r   imreadshapelencv2cvtColorCOLOR_GRAY2RGBnparray)img_fileimg r   J/var/www/html/scripts/venv/lib/python3.10/site-packages/easyocr/imgproc.py	loadImage   s   
(
r   g
ףp=
?gv/?gCl?gZd;O?gy&1?g?c                 C   sx   |   tj}|tj|d d |d d |d d gtjd8 }|tj|d d |d d |d d gtjd }|S )Nr        o@   r   dtype)copyastyper   float32r   in_imgmeanvariancer   r   r   r   normalizeMeanVariance   s   22r    c                 C   s:   |   }||9 }||7 }|d9 }t|ddtj}|S )Nr   r      )r   r   clipr   uint8r   r   r   r   denormalizeMeanVariance   s   r$   r   c                 C   s   | j \}}}|t|| }||kr|}|t|| }t|| t|| }	}
tj| |
|	f|d}|	|
}}|	d dkrD|	d|	d   }|
d dkrR|
d|
d   }tj|||ftjd}||d|	d|
d d f< ||}	}
t|
d t|	d f}|||fS )N)interpolation    r   r   r   )r   maxintr	   resizer   zerosr   )r   square_sizer%   	mag_ratioheightwidthchanneltarget_sizeratiotarget_htarget_wproc
target_h32
target_w32resizedsize_heatmapr   r   r   resize_aspect_ratio%   s"   


r9   c                 C   s,   t | ddd t j} t| tj} | S )Nr   r   r!   )r   r"   r   r#   r	   applyColorMapCOLORMAP_JET)r   r   r   r   cvt2HeatmapImgC   s   r<   )r   r   )r   )__doc__numpyr   skimager   r	   r   r    r$   r9   r<   r   r   r   r   <module>   s    
	

	