o
'j< @ s d dl Z d dlZe je jeZeje ejd e je j ed de j
d< d dl Z d dl Z
d dlZd dlZd dlm mZ d dlmZmZ d dlmZ d dlmZ d dlmZm Z d d l m Z d d
l m Z e Z!d d Z"G d
d de#Z$dd Z%e&dkre%e dS dS ) Nz../..Z auto_growthZFLAGS_allocator_strategy)create_operators transform)build_post_process)
get_logger)get_image_file_listcheck_and_read)draw_rectangle)
parse_argsc C s dd| j ii}dd| j | j gii}d| jdvrg dng d| jdvr'g d ng dd
d d i}d
d i}ddddgii}| jdvrJ|||||g}|S |||||g}|S )NZResizeTableImagemax_lenZPaddingTableImagesizeZNormalizeImageZ TableMaster)g Zd;O?gy&1?g?) ?r
r
)g
ףp=
?gv/?gCl?z1./255.Zhwc)ZstdmeanscaleorderZ
ToCHWImageZKeepKeysZ keep_keysimageshape)Z
table_max_lentable_algorithm)argsZ resize_opZpad_opZ normalize_opZ to_chw_opZ keep_keys_oppre_process_list r n/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddleocr/ppstructure/table/predict_structure.pybuild_pre_process_list' s2
r c @ s e Zd Zdd Zdd ZdS )TableStructurerc C s || _ |j| _t|}|jdvrd|j|jd}n d|jd|jd}t|| _t|| _ t
|dt \| _
| _| _| _|jrgdd l}t }t
}|jd|jd d
d | j|d |jr[|nd g d dt d | _d S d S )
Nr ZTableLabelDecode)namecharacter_dict_pathmerge_no_span_structureZTableMasterLabelDecodepad)r r Z box_shaper tabler Zdynamic)Zpreprocess_timeZinference_timeZpostprocess_time) Z
model_nameZmodel_precisionZ
batch_sizeZ
data_shapeZ save_pathZinference_configZpidsZ process_nameZgpu_idsZ time_keysZwarmuplogger)r use_onnxr r Ztable_char_dict_pathr r
preprocess_opr postprocess_oputilityZcreate_predictorr predictor input_tensoroutput_tensorsconfig benchmarkauto_logosgetpidZget_infer_gpuidZ
AutoLogger precisionZuse_gpuautolog)selfr r Zpostprocess_paramsr* pidZgpu_idr r r __init__F sH
zTableStructurer.__init__c C s t }| jjr| jj | }d|i}t|| j}|d }|d u r&dS t j
|dd}| }| jjr;| jj | j rOi }||| j
j< | j| j|}n&| j
| | j g }| jD ] }| }|| q_| jjru| jj i } |d | d< |d | d< t j
|d dd}
| | |
g} | d d } | d
d }
| d } g d | g d } t | }| jjr| jjjd
d | |
f|fS )Nr r )Nr )Zaxisr Zstructure_probsZ loc_predsZstructure_batch_listZbbox_batch_list)zzzzzT)stamp)timer r) r. timesstartcopyr r" npZ expand_dimsr3 r! r& r r% runr' Z
copy_from_cpuZ copy_to_cpuappendr# end)r/ imgZ starttimeZori_imdataZ
input_dictZoutputsZ
output_tensoroutputpredsZ
shape_listZ post_resultstructure_str_list bbox_listelapser r r __call__q sT
zTableStructurer.__call__N)__name__
__module__ __qualname__r1 rC r r r r r E s +r c C s t | j}t| }d}d}tj| jdd ttj| jdddd}|D ]}t |\}}} |s5t
|}|d u rBt
d| q%||\}
} |
\} }
t|
}t
d | | |d
| | t|
dkryt|
d d kryt||
}nt||
}tj| jtj|}t
|| t
d | |dkr|| 7 }|d
7 }t
d|| q%W d n1 sw Y | jr|j d S d S )Nr T)exist_okz infer.txtwzutf-8)modeencodingzerror in loading image:{}zresult: {}, {}zresult: {}, {}
zsave vis result to {}r zPredict time of {}: {}) r Z image_dirr r+ makedirsr> openpathjoinr cv2Zimreadr infoformatjsondumpstolistwritelenr r$ Z
draw_boxesbasenameZimwriter) r. report)r Zimage_file_listZtable_structurercount
total_timeZf_wZ
image_filer< flag_Z
structure_resrB r@ rA Z
bbox_list_strZ
img_save_pathr r r main sV
r^ __main__)'r+ sysrN dirnameabspath__file____dir__r: insertrO environrP numpyr8 r4 rS Ztools.infer.utilityZinferr$ Z
ppocr.datar r Zppocr.postprocessr Zppocr.utils.loggingr Zppocr.utils.utilityr r Zppocr.utils.visualr Zppstructure.utilityr r r objectr r^ rD r r r r s0
\(