o 'j2 @sddlZddlZddlZejejeZejeej dejej eddej d<ddl Z ddl Z ddlZddlZddlZddlmZddlmZmZddlmZddlmZmZdd lmZdd lmZdd l m!Z!m"Z"dd l#m$Z$m%Z%eZ&Gd dde'Z(dddZ)ddZ*e+dkre$Z,e,j-rgZ.e,j/Z/e0e/D]#Z1ej2dgej3d4e1d4dgZ5ej6e5ej7ej7dZ8e.e8qe.D]Z8e89qdSe*e,dSdS)Nz../Z auto_growthZFLAGS_allocator_strategy)deepcopy)get_image_file_listcheck_and_read) get_logger)draw_ser_resultsdraw_re_results) TextSystem)LayoutPredictor) TableSystemto_excel) parse_argsdraw_structure_resultc@seZdZddZdddZdS) StructureSystemcCs|j|_|j|_d|_|jrddl}|jdd|_|jdkrs|js'tt j |j dkr9|j dkr9d|_ t dd|_d|_d|_d|_|j rUt||_|j rUt||_|jrq|jdurjt||jj|jj|_dSt||_dSdS|jdkrdd lm}|||_dSdS) NrZtext_image_orientation)Z model_name structureFTzAWhen args.layout is false, args.ocr is automatically set to falsekie)SerRePredictor)moderecoveryimage_orientation_predictorimage_orientation paddleclasZ PaddleClasZshow_logloggersetLevelloggingINFOlayoutZocrwarningZ drop_scorelayout_predictor text_system table_systemr rtabler Z text_detectorZtext_recognizerZ(ppstructure.kie.predict_kie_token_ser_rer kie_predictor)selfargsrrr$e/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddleocr/ppstructure/predict_system.py__init__+sH        zStructureSystem.__init__Frc& Csrddddddddd}t}|jdurJt}|jj|d}t|}|ddd} tjtjtjd} | | vr@t|| | }t} | ||d<|j dkr| } |j duri| |\} }|d|7<n| j dd \}}t dd d g} g}| D]}d }|d dur|d \}}}}t|t|t|t|f\}}}}| ||||ddf}n dd||f\}}}}| }|dd kr|jdur|||\}}|d |d 7<|d|d7<|d|d7<|d|d7<n|jdurz|jr!tj| j | jd}||||||ddf<||\}}}n||\}}}|d|d7<|d|d7<gd}g}t||D]1\}} | \}!}"|D]}#|#|!vr_|!|#d }!qR|jsk|||g7}||!t|"|dqH||d||||g|||dq}t}$|$||d<||fS|j dkr||\}%}||d<||d<|%d|fSdS)Nr)rrr table_matchdetrecrall)Z input_dataZ label_names)Z90Z180Z270rrrr )bboxlabelr,r-r'matchr(r))dtype)r1zzzzzzz z z z zz)textZ confidenceZ text_region)typer,imgresimg_idxr*r)NN)timerZpredictnextcv2ZROTATE_90_COUNTERCLOCKWISEZ ROTATE_180ZROTATE_90_CLOCKWISErotatercopyrshapedictintrrrnpZonesr0zipreplaceappendfloattolistlowerr!)&r"r4Zreturn_ocr_result_in_tabler6 time_dictstartZticZ cls_resultZcls_resZangleZcv_rotate_codeZtocZori_imZ layout_resZelapsehwZres_listregionr5x1y1Zx2y2roi_imgZtable_time_dictZwht_imZ filter_boxesZfilter_rec_resZ ocr_time_dictZ style_tokenboxZrec_resZrec_strZrec_conftokenendZre_resr$r$r%__call__Rs      $               zStructureSystem.__call__N)Fr)__name__ __module__ __qualname__r&rRr$r$r$r%r*s'rc Cs"tj||}tj|ddt|}ttj|d|dddf}|D]Z}|d}|dt ||d  d krct |d d krcd |d vrctj|d|d|} t |d d | q$|d  dkr~tj|d|d|} t| |q$WddS1swYdS)NTexist_okz res_{}.txtrIutf8encodingr4z{} r3r r5rhtmlz {}_{}.xlsxr,Zfigurez {}_{}.jpg)ospathjoinmakedirsropenformatpopwritejsondumpsrElenr r9imwrite) r5 save_folderimg_namer6Zexcel_save_folderZres_cpfrJrNZ excel_pathZimg_pathr$r$r%save_structure_ress@   "rkc CsLt|j}|}||jd|j}|js't|}tj|j |j }tj |ddt |}t |D]s\}}td|||t|\}}} tj|dd} |jr|jr| rddlm} tj |j ddtj|j d| } | |} | | | td| q/|s| st|}| s|durtd |q/|g}n|}g}t |D]\}}|||d \}}tj|| d |}tj tj|| dd|j d kr|gkrt|||j}t||| |nP|j d kr6|jj durt!|||jd}nt"|||jd}t#tj|| d|ddd}d|t$j%d|idd}|&|Wdn 1s1wY|gkrIt'||td||jri|gkriddl(m)}m*}|j+\}}}|||}||7}q|jr|gkrz ||||| Wnt,y}ztd||WYd}~q/d}~wwtd|dq/dS)NTrVz [{}/{}] {}.r) Converterz {}_api.docxzdocx save to {}zerror in loading image:{})r6z show_{}.jpgrr)Z font_pathzres_{}_kie.txtrIrXrYz{} {} Zocr_infoF) ensure_asciizresult save to {})sorted_layout_boxesconvert_info_docxz.error in layout recovery image:{}, err msg: {}zPredict time : {:.3f}sr*)-rZ image_dir process_idtotal_process_numZuse_pdf2docx_apirr\r]r^outputrr_rf enumeraterinforarbasenamesplitrZpdf2docx.converterrmconvertcloser9Zimreaderrorr Z vis_font_pathrkr!Z predictorrrr`rdrercrgZ$ppstructure.recovery.recovery_to_docrorpr< Exception)r#Zimage_file_listZ structure_sysrhZimg_numiZ image_filer4Zflag_gifZflag_pdfrirmZ docx_fileZcvZimgsZall_resindexr5rFZ img_save_pathZdraw_imgrjres_strrorprHrI_exr$r$r%mains              r__main__z-uz--process_id={}z --use_mp={}F)stdoutstderr)r):r\sys subprocessr]dirnameabspath__file____dir__rBinsertr^environr9rdnumpyr?r7rr;rZppocr.utils.utilityrrZppocr.utils.loggingrZppocr.utils.visualrrZtools.infer.predict_systemrZ!ppstructure.layout.predict_layoutr Zppstructure.table.predict_tabler r Zppstructure.utilityr r robjectrrkrrSr#Zuse_mpZp_listrrrangerq executableargvracmdPopenrpwaitr$r$r$r%sT        V