o èeU!ã @sddlZddlZej ej e¡¡Zej e¡ej dej ej  ed¡¡¡ej dej ej  ed¡¡¡dej d<ddl Z ddl Z ddl Z ddlZddlZddlmmZddlmmZddlmmZddlmZddlmZmZddlmZdd lm Z dd l!m"Z"dd l#m$Z$ddl%m&m'Z(eƒZ)d d „Z*Gdd„de+ƒZ,dd„Z-dd„Z.e/dkrþe$ƒZ0e0j1røddl2Z2gZ3e0j4Z4e5e4ƒD]#Z6ej7dgej8d 9e6¡d 9d¡gZ:e2j;e:ej¡qïdSe.e0ƒdSdS)éNz..z../..Z auto_growthZFLAGS_allocator_strategy)Ú sorted_boxes)Úget_image_file_listÚcheck_and_read)Ú get_logger)Ú TableMatch)ÚTableMasterMatcher)Ú parse_argscCs‚|\}}}}|\}}} ||} ||} ||} ||} | dkr!| nd}| |kr)| n|}| dkr1| nd}| |kr9| n|}||||fS)Nr©)ZpixÚdet_boxÚshapeÚx0Úy0Úx1Úy1ÚhÚwÚcZtmp_x0Ztmp_x1Ztmp_y0Ztmp_y1Zx0_Zx1_Zy0_Zy1_r r úYD:\Projects\ConvertPro\env\Lib\site-packages\paddleocr/ppstructure/table/predict_table.pyÚexpand+s   rc@s0eZdZd dd„Zd dd„Zdd„Zd d „ZdS) Ú TableSystemNcCsÂ||_|js t tj¡d}|jr|j}d|_|dur#t t   |¡¡n||_ |dur2t   t   |¡¡n||_|r:d|_t |¡|_|jdvrJtƒ|_ntdd|_t |dt¡\|_|_|_|_dS)NFT)Z TableMaster)Zfilter_ocr_resultÚtable)ÚargsZshow_logÚloggerÚsetLevelÚloggingÚINFOÚ benchmarkÚ predict_detZ TextDetectorÚcopyÚdeepcopyÚ text_detectorÚ predict_recZTextRecognizerÚtext_recognizerÚpredict_strtureZTableStructurerÚtable_structurerZtable_algorithmrÚmatchrÚutilityZcreate_predictorZ predictorZ input_tensorZoutput_tensorsÚconfig)Úselfrr r"Z benchmark_tmpr r rÚ__init__;s8 ÿÿÿÿÿÿ    ÿzTableSystem.__init__FcCsÔtƒ}ddddddœ}t ¡}| t |¡¡\}}|d ¡|d<||d<| t |¡¡\}} } } | |d<| |d<|rC||d<| |d <t ¡} | ||| ¡} t ¡}|| |d <| |d <t ¡}|||d <||fS) Nr)ÚdetÚrecrÚallr%éÚ cell_bboxrr*r+ZboxesÚrec_resr%Úhtmlr,)ÚdictÚtimeÚ _structurerrÚtolistÚ_ocrr%)r(ÚimgZreturn_ocr_result_in_tableÚresultZ time_dictÚstartÚ structure_resÚelapseÚdt_boxesr/Ú det_elapseÚ rec_elapseZticÚ pred_htmlZtocÚendr r rÚ__call__Rs, ÿ  zTableSystem.__call__cCs| t |¡¡\}}||fS)N)r$rr)r(r6r9r:r r rr3lszTableSystem._structurecCsŽ|jdd…\}}| t |¡¡\}}t|ƒ}g}|D]I}td|dd…df ¡dƒ}t||dd…df ¡dƒ} td|dd…df ¡dƒ} t||dd…df ¡dƒ} || | | g}| |¡qt  |¡}t   d  t |ƒ|¡¡|dur{dSg} tt |ƒƒD]+} || }td||jƒ\}}}}|t|ƒt|ƒ…t|ƒt|ƒ…dd…f}|  |¡qƒ| | ¡\}}t   d  t |ƒ|¡¡||||fS)Nérr-zdt_boxes num : {}, elapse : {}©NNzrec_res num : {}, elapse : {})r r rrrÚmaxÚminÚappendÚnpÚarrayrÚdebugÚformatÚlenÚrangerÚintr")r(r6rrr;r<Zr_boxesÚboxZx_minZx_maxZy_minZy_maxZ img_crop_listÚir r r rrZ text_rectr/r=r r rr5ps8   ÿ* ÿ zTableSystem._ocrrB)F)Ú__name__Ú __module__Ú __qualname__r)r@r3r5r r r rr:s    rcCsddlm}| ||¡dS)Nr)Ú tablepyxl)rRZdocument_to_xl)Z html_tableÚ excel_pathrRr r rÚto_excels rTcCs¢t|jƒ}||jd|j…}tj|jddt|ƒ}t|ƒ}t tj   |jd¡ddd}|  d¡|  d¡|  d ¡|  d ¡|  d ¡|  d ¡|  d ¡|  d¡|  d¡t |ƒD]Ý\}}t d |||¡¡t|ƒ\}}} tj   |jtj  |¡ d¡dd¡} |s‹t |¡}|dur˜t d |¡¡q\t ¡} ||ƒ\} } | d} t | ¡t| | ƒt d | ¡¡t ¡| }t d |¡¡t| dƒdkrát| ddƒdkrát || d¡}nt || d¡}tj   |jtj  |¡¡}t ||¡|  d ¡|  dtj  |¡›d¡|  d|›d¡|  d|  dd ¡ d!d ¡d"¡|  dtj  |¡›d¡|  d¡q\|  d#¡| ¡|j rO|j!j" #¡dSdS)$NT)Úexist_okz show.htmlrzutf-8)ÚmodeÚencodingz z zEz z zzz z [{}/{}] {}Ú.rz.xlsxzerror in loading image:{}r0zexcel saved to {}zPredict time : {:.3f}sr.éz z
img name zori imagetable htmlcell box
z
z		z
Úz
z
z )$rZ image_dirÚ process_idÚtotal_process_numÚosÚmakedirsÚoutputrrJÚopenÚpathÚjoinÚwriteÚ enumeraterÚinforIrÚbasenameÚsplitÚcv2ZimreadÚerrorr2rTr#Zdraw_rectangler&Z draw_boxesZimwriteÚreplaceÚcloserr$ZautologÚreport)rZimage_file_listZ table_sysZimg_numZf_htmlrNZ image_filer6ÚflagÚ_rSZ starttimeZpred_resr>r:Z img_save_pathr r rÚmain”sˆ ÿ  ÿ      ÿ     ÿÿÿ   ÿÿþÿ  ÿroÚ__main__z-uz--process_id={}z --use_mp={}F)ÚstdoutÚstderr)?r]ÚsysraÚdirnameÚabspathÚ__file__Ú__dir__rEÚinsertrbÚenvironrhrrÚnumpyrFr2Ztools.infer.predict_recZinferr!Ztools.infer.predict_detrZtools.infer.utilityr&Ztools.infer.predict_systemrZppocr.utils.utilityrrZppocr.utils.loggingrZppstructure.table.matcherrZ$ppstructure.table.table_master_matchrZppstructure.utilityrZ#ppstructure.table.predict_structurerZpredict_structurer#rrÚobjectrrTrorOrZuse_mpÚ subprocessZp_listr\rKr[Ú executableÚargvrIÚcmdÚPopenrqÚpÚwaitr r r rÚsZ         UB þ  ÿ ð