o èe:ã @s ddlZddlZej ej e¡¡Zej e¡ej dej ej  ed¡¡¡dej d<ddl Z ddl Z ddlZddlmmZddlmZmZddlmZddlmZddlmZmZdd lmZdd lmZeƒZ Gd d „d e!ƒZ"d d„Z#e$dkr„e#eƒƒdSdS)éNz../..Z auto_growthZFLAGS_allocator_strategy)Úcreate_operatorsÚ transform)Úbuild_post_process)Ú get_logger)Úget_image_file_listÚcheck_and_read)Ú parse_args)ÚPicoDetPostProcessc@seZdZdd„Zdd„ZdS)ÚLayoutPredictorcCs†ddddgiidgd¢gd¢dd d œid did d dgiig}d|j|j|jdœ}t|ƒ|_t|ƒ|_t |dt ¡\|_ |_ |_ |_ dS)NZResizeÚsizei i`ZNormalizeImage)gZd;ßOÍ?gyé&1¬Ì?gÍÌÌÌÌÌÌ?)g ×£p= ß?gÉv¾Ÿ/Ý?g–C‹lçûÙ?z1./255.Zhwc)ZstdÚmeanÚscaleÚorderZ ToCHWImageZKeepKeysZ keep_keysÚimager )ÚnameÚlayout_dict_pathZscore_thresholdZ nms_thresholdZlayout)rZlayout_score_thresholdZlayout_nms_thresholdrÚ preprocess_oprÚpostprocess_opÚutilityZcreate_predictorÚloggerÚ predictorÚ input_tensorZoutput_tensorsÚconfig)ÚselfÚargsZpre_process_listZpostprocess_params©rú[D:\Projects\ConvertPro\env\Lib\site-packages\paddleocr/ppstructure/layout/predict_layout.pyÚ__init__'s6ÿÿüÿÿÿÿóü   ÿzLayoutPredictor.__init__c Cs| ¡}d|i}t||jƒ}|d}|durdStj|dd}| ¡}d\}}t ¡}|j |¡|j  ¡gg}}|j  ¡} t t | ƒdƒ} t | ƒD]} | |j | | ¡ ¡¡| |j | | | ¡ ¡¡qLt||d}| |||¡} t ¡|}| |fS)Nrr)Nr)Zaxis)réé)ZboxesZ boxes_num)ÚcopyrrÚnpZ expand_dimsÚtimerZ copy_from_cpurÚrunZget_output_namesÚintÚlenÚrangeÚappendZget_output_handleZ copy_to_cpuÚdictr) rÚimgZori_imÚdataÚpredsÚelapseZ starttimeZ np_score_listZ np_boxes_listZ output_namesZnum_outsZout_idxZ post_predsrrrÚ__call__Fs<      þÿþ  zLayoutPredictor.__call__N)Ú__name__Ú __module__Ú __qualname__rr-rrrrr &s r c Cs¨t|jƒ}t|ƒ}d}d}d}|D]@}t|ƒ\}}} |s!t |¡}|dur.t d |¡¡q||ƒ\} } t d | ¡¡|dkrD|| 7}|d7}t d || ¡¡qdS)Nré2zerror in loading image:{}z result: {}rzPredict time of {}: {}) rZ image_dirr rÚcv2ZimreadrÚinfoÚformat) rZimage_file_listZlayout_predictorÚcountÚ total_timeZrepeatsZ image_filer)ÚflagÚ_Z layout_resr,rrrÚmainis&   ñr9Ú__main__)%ÚosÚsysÚpathÚdirnameÚabspathÚ__file__Ú__dir__r'ÚinsertÚjoinÚenvironr2Únumpyr!r"Ztools.infer.utilityZinferrZ ppocr.datarrZppocr.postprocessrZppocr.utils.loggingrZppocr.utils.utilityrrZppstructure.utilityrZpicodet_postprocessr rÚobjectr r9r.rrrrÚs,        Cÿ