o
    'Õj  ã                	   @   sø   d dl Z d dlZe j e j e¡¡Zej e¡ ej d e j e j 	ed¡¡¡ d dl
Z
d dlZd dlZd dlmZ d dlmZ d dlmZ d dlmZ d dlmZ eƒ Zdd	„ Zd
d„ Zdd„ Zdd„ Zdd„ Zedkrzeƒ Zeejejeƒ dS dS )é    Nz../..)Útqdm)ÚTEDS)ÚTableSystem)Ú	init_args)Ú
get_loggerc                  C   s   t ƒ } | jdtd |  ¡ S )Nz	--gt_path)Útype)r   Úadd_argumentÚstrÚ
parse_args)Úparser© r   úg/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddleocr/ppstructure/table/eval_table.pyr
   "   s   r
   c                 C   sv   i }t j | ¡s
|S t| dd!}| ¡ }|D ]}| ¡  d¡}|\}}|||< qW d   ƒ |S 1 s4w   Y  |S )Nzutf-8)Úencodingú	)ÚosÚpathÚexistsÚopenÚ	readlinesÚstripÚsplit)Ztxt_pathZpred_html_dictÚfÚlinesÚlineÚimg_nameÚ	pred_htmlr   r   r   Úload_txt(   s   
ý
þúr   c                 C   s$   i }t j | ¡rt t| dƒ¡}|S )NÚrb)r   r   r   ÚpickleÚloadr   )r   Údatar   r   r   Úload_result5   s   r!   c                 C   sN   t | ƒ}| |¡ t| dƒ}t ||¡ W d   ƒ d S 1 s w   Y  d S )NÚwb)r!   Úupdater   r   Údump)r   r    Zold_datar   r   r   r   Úsave_result<   s
   
"ÿr%   c                 C   sV  t j|jdd t|ƒ}t| ƒ}tt j |jd¡ƒ}tt j |jd¡ƒ}g }g }t| 	¡ ƒD ]a\}	}
t
 t j ||	¡¡}|	|vrZ| |¡\}}}}||g||	< tt j |jd¡|ƒ |	|vrt| |¡\}}|||	< tt j |jd¡|ƒ ||	 \}}||	 }| |||¡}| |¡ | |
¡ q.tdd}| ||¡}t d t|ƒt|ƒ ¡¡ d S )NT)Úexist_okz
ocr.picklezstructure.pickleé   )Zn_jobszteds: {})r   ÚmakedirsÚoutputr   r   r!   r   Újoinr   ÚitemsÚcv2ZimreadZ_ocrr%   Ú
_structureÚmatchÚappendr   Zbatch_evaluate_htmlÚloggerÚinfoÚformatÚsumÚlen)Úgt_pathZimg_rootÚargsZtext_sysZgt_html_dictZ
ocr_resultZstructure_resultZ
pred_htmlsZgt_htmlsr   Zgt_htmlZimgZdt_boxesZrec_resÚ_Zstructure_resr   ZtedsZscoresr   r   r   ÚmainC   s:   ÿÿ

 r8   Ú__main__) r   Úsysr   ÚdirnameÚabspathÚ__file__Ú__dir__r/   Úinsertr*   r,   r   Zpaddler   Zppstructure.table.table_metricr   Zppstructure.table.predict_tabler   Zppstructure.utilityr   Zppocr.utils.loggingr   r0   r
   r   r!   r%   r8   Ú__name__r6   r5   Z	image_dirr   r   r   r   Ú<module>   s.    (þ