o 'Õjtqã @s ddlZddlZddlmZej ej e¡¡Zej e¡ej dej ej ed¡¡¡dejd<ddl Z ddlZddlZddlZddlZddlZddlmmZddlmZddlmZddlmZmZeƒZGd d „d eƒZ dd„Z!e"d krƒe!e #¡ƒdSdS)éN)ÚImagez../..Zauto_growthZFLAGS_allocator_strategy)Úbuild_post_process)Ú get_logger)Úget_image_file_listÚcheck_and_readc@speZdZdd„Zdd„Zdd„Zdd„Zd d „Zdd„Z ddd„Z dd„Z dd„Zdd„Zdd„Z dd„ZdS)ÚTextRecognizercCs>dd„|j d¡Dƒ|_|j|_|j|_d|j|jdœ}|jdkr*d|j|jdœ}n¨|jdkr8d |j|jdœ}nš|jd krFd|j|jdœ}nŒ|jdkrTd |j|jdœ}n~|jdkrbd|j|jdœ}np|jdkrpd|j|jdœ}nb|jdkr~d|j|jdœ}nT|jdkrŒd|j|jdœ}nF|jdkr›d |j|jddœ}n7|jdkr¨dd|jdœ}n*|jdkr·d|j|jddœ}n|jdkrÁddi}n|jd krÒ|j|_d!|j|jdœ}t|ƒ|_ t |d"t¡\|_ |_|_|_|j|_|j|_|jrd#dl}t ¡}t ¡}|jd"|j|jd$d|j|d|jr|ndgd%¢d#td&|_dSdS)'NcSsg|]}t|ƒ‘qS©)Úint)Ú.0Úvrrúb/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddleocr/tools/infer/predict_rec.pyÚ (sz+TextRecognizer.__init__..ú,ZCTCLabelDecode)ÚnameÚcharacter_dict_pathÚuse_space_charÚSRNZSRNLabelDecodeÚRAREZAttnLabelDecodeÚNRTRZNRTRLabelDecodeÚSARZSARLabelDecodeÚ VisionLANZ VLLabelDecodeÚViTSTRZViTSTRLabelDecodeÚABINetZABINetLabelDecodeÚSPINZSPINLabelDecodeÚ RobustScannerT)rrrZ rm_symbolÚRFLZRFLLabelDecodeÚSATRNZSATRNLabelDecodeÚPRENrZPRENLabelDecodeÚCANZCANLabelDecodeZrecrZdynamic)Zpreprocess_timeZinference_timeZpostprocess_time)Z model_nameZmodel_precisionZ batch_sizeZ data_shapeZ save_pathZinference_configZpidsZprocess_nameZgpu_idsZ time_keysÚwarmupÚlogger)Úrec_image_shapeÚsplitÚ rec_batch_numÚ rec_algorithmZrec_char_dict_pathrZrec_image_inverseÚinverserÚpostprocess_opÚutilityZcreate_predictorr Ú predictorÚinput_tensorÚoutput_tensorsÚconfigÚ benchmarkÚuse_onnxÚauto_logÚosÚgetpidZget_infer_gpuidZ AutoLoggerÚ precisionZuse_gpuÚautolog)ÚselfÚargsZpostprocess_paramsr.ÚpidZgpu_idrrrÚ__init__'sÀý ý ý ý ý ý ý ý ý ü ý ü ý ÿòüzTextRecognizer.__init__cCsb|j\}}}|jdks|jdkrbt |tj¡}t t |¡¡}|jdkr.| ||gtj ¡}n | ||gtj¡}t |¡}t |d¡}| d¡}|jdkrV| tj¡d}|S| tj¡dd}|S|jdkr•t |tj¡}tj |||ftjd }| d ¡}|d}|tjdd…f}|d8}|d}|S||jd ksžJ‚t||ƒ}|jrÁ|jjdd…d} t| tƒr·n | durÁ| dkrÁ| }|jdd …\} } | t| ƒ}t ||¡|krÜ|}n tt ||¡ƒ}|jdkrû||jd krö|jd }|jd }t |||f¡}| d ¡}| d¡d}|d8}|d}tj|||ftjd} || dd…dd…d|…f<| S)Nrréÿÿÿÿ©éréçào@g`@çð?r©Ú interpolationÚfloat32éÿçà?r9érr©Zdtype)r!r$Úcv2ÚcvtColorÚCOLOR_BGR2GRAYrZ fromarrayÚnpÚuint8ÚresizeZBICUBICZLANCZOSÚarrayÚexpand_dimsÚ transposeÚastyper?ÚINTER_CUBICÚnewaxisÚshaper r-r)Ú isinstanceÚstrÚfloatÚmathÚceilÚzeros)r3ÚimgÚmax_wh_ratioÚimgCÚimgHÚimgWZ image_pilÚnorm_imgÚ resized_imageÚwÚhÚratioZ resized_wÚ padding_imrrrÚresize_norm_img–sf ÿ ÿ zTextRecognizer.resize_norm_imgcCsX|\}}}|dd…dd…ddd…f}tj|||ftjd}| d¡}| d¡d}|S)Nr7r=r?r8r@©rDrIÚINTER_LINEARrMrL©r3rWÚimage_shaperYrZr[r]rrrÚresize_norm_img_vlÎs ÿ z!TextRecognizer.resize_norm_img_vlcCs|\}}}t ||f¡}|jd}|jd}||dkr't ||d|f¡} n*||dkr8t ||d|f¡} n||dkrIt ||d|f¡} nt |||f¡} t | ¡} t | tj¡} | |dd…d| jd…f<|dd…dd…tjf}|j\}}} d} t || ||f¡ tj¡S)Nrr:r9rB)rGrVrPrDrIZasarrayrErFrOÚreshaperMr?)r3rWrfrYrZr[Z img_blackZim_heiZim_widZimg_newZimg_npÚrowÚcolÚcrrrÚresize_norm_img_srnØs$ z"TextRecognizer.resize_norm_img_srnc Cs|\}}}t|d|dƒ}t td|ƒ¡ |df¡ d¡}t td|ƒ¡ |df¡ d¡} t d||f¡} t | d¡ dd||g¡}t |d|ddg¡ d¡dg}t | d¡ dd||g¡}t |d|ddg¡ d¡dg}|tj dd…f}| tj dd…f} || ||gS)Nérr:Úint64r7r?geÍÍÁ)r rGrJÚrangerhrMÚonesZtriuZtileZtrilrO) r3rfÚ num_headsÚmax_text_lengthrYrZr[Zfeature_dimÚencoder_word_posÚ gsrm_word_posZgsrm_attn_bias_dataÚgsrm_slf_attn_bias1Úgsrm_slf_attn_bias2rrrÚsrn_other_inputsòsH ÿÿÿÿ ÿ þ þ ÿ þ þþzTextRecognizer.srn_other_inputsc Csr| ||¡}|tjdd…f}| |||¡\}}}} | tj¡}| tj¡} | tj¡}| tj¡}||||| fS)N)rlrGrOrwrMr?rn) r3rWrfrqrrr\rsrtrurvrrrÚprocess_image_srns ÿÿz TextRecognizer.process_image_srnçÐ?cCsH|\}}}}|jd}|jd} d} td|ƒ}| t|ƒ}t ||¡} | |dkr3t| |ƒ|} |dur|jdkr›t |¡}t |¡}|||g}"|j!rWi}#||#|j"j#<|j$ %|j&|#¡}$|$d}%n|j$ '¡}&t t|&ƒƒD]}'|j$ (|&|'¡}(|( )|"|'¡qb|j$ %¡g}$|j&D]})|) *¡}*|$ |*¡q|j r–|j j ¡|$d}%nË|jdkrt | ¡} t |!¡}!|| |!g}"|j!rÇi}#||#|j"j#<|j$ %|j&|#¡}$|$}%nŸ|j$ '¡}&g}(t t|&ƒƒD]}'|j$ (|&|'¡}+|+ )|"|'¡|( |+¡qÔ|(|_"|j$ %¡g}$|j&D]})|) *¡}*|$ |*¡qù|j r|j j ¡|$}%nS|j!r,i}#||#|j"j#<|j$ %|j&|#¡}$|$d}%n:|j" )|¡|j$ %¡g}$|j&D]})|) *¡}*|$ |*¡q<|j rS|j j ¡t|$ƒdkr]|$}%n|$d}%|j$ ,¡| -|%¡},t t|,ƒƒD] }-|,|-||| |-<qq|j r‹|j jj.ddq?|t ¡|fS)Nr:rÚgrrrBr9r<)Zaxisrméé)ZSVTRr)rrrrrry)r}é(rnrr?rCé$ZpredictT)Ústamp)/ÚlenÚappendrPrSrGZargsortrJr#Útimer,r2ÚtimesÚstartror|r$r!r{rrOrKrxr†rgr…r‡rMrŠrprbZconcatenateÚcopyrr-r)rr(Úrunr*Zget_input_namesZget_input_handleZ copy_from_cpuZcopy_to_cpur?Ztry_shrink_memoryr&Úend).r3Úimg_listZimg_numZ width_listrWÚindicesÚrec_resZ batch_numÚstZ beg_img_noZ end_img_noZnorm_img_batchZencoder_word_pos_listZgsrm_word_pos_listZgsrm_slf_attn_bias1_listZgsrm_slf_attn_bias2_listZvalid_ratiosrYrZr[rXÚinor_r^Zwh_ratior\r‰r~Zword_positions_listZword_positionsZnorm_image_maskZ word_labelZnorm_img_mask_batchZword_label_listZinputsZ input_dictZoutputsÚpredsZinput_namesÚir)Z output_tensorÚoutputZinput_tensor_iZ rec_resultZrnorrrÚ__call__‰sê ÿ ÿÿÿÿ ý ÿ ÿ ÿÿû ÿ ÿ ÿþ ÿ ÿ ÿ ÿ ÿ ÿ ÿ €zTextRecognizer.__call__N)ry)Ú__name__Ú __module__Ú__qualname__r6rbrgrlrwrxrr…r†r‡rŠr¡rrrrr&so8 ÿ#rc CsTt|jƒ}t|ƒ}g}g}t d¡|jr4tj ddgd¢¡ tj ¡}tdƒD]}||gt|j ƒƒ}q'|D]'}t|ƒ\}} } | sFt |¡}|durSt d |¡¡q6| |¡| |¡q6z||ƒ\}} Wn!ty‡}zt t ¡¡t |¡tƒWYd}~nd}~wwtt|ƒƒD]} t d || || ¡¡qŽ|jr¨|j ¡dSdS)NzIn PP-OCRv3, rec_image_shape parameter defaults to '3, 48, 320', if you are using recognition model with PP-OCRv2 or an older version, please set --rec_image_shape='3,32,320rr@)é0i@rBr9zerror in loading image:{}zPredicts of {}:{})rZ image_dirrr ÚinforrGÚrandomÚuniformrMrHror r#rrDZimreadÚformatr’Ú ExceptionÚ tracebackÚ format_excÚexitr‘r,r2Úreport)r4Zimage_file_listZtext_recognizerZvalid_image_file_listr™rWrŸÚresZ image_fileÚflagr‰r›ÚErrrrÚmain|sF ÿ €ýÿÿr²Ú__main__)$r/ÚsysZPILrÚpathÚdirnameÚabspathÚ__file__Ú__dir__r’ÚinsertÚjoinÚenvironrDÚnumpyrGrTr“r«ZpaddleZtools.infer.utilityZinferr'Zppocr.postprocessrZppocr.utils.loggingrZppocr.utils.utilityrrr Úobjectrr²r¢Ú parse_argsrrrrÚs6 Z'ÿ