o 0j<(@sddlmZddlZddlmZddlmZddlm Z m Z ddl m Z m Z mZmZmZmZmZmZmZmZmZddlmZd d lmZd d lmZd d lmZmZddl m!Z!m"Z"m#Z#m$Z$ddl%m&Z&e drnddl'm(Z(gdZ)ddZ*e dGdddeZ+GdddeZ,dS))OptionalN)Image)logging)class_requires_depsis_dep_available) ARABIC_FONT CYRILLIC_FONTDEVANAGARI_FONTEL_FONT KANNADA_FONT KOREAN_FONT LATIN_FONT SIMFANG_FONT TAMIL_FONT TELUGU_FONTTH_FONT) FuncRegisterImageBatchSampler ReadImage)RunnerPredictorTransformersPredictor)CTCLabelDecodeOCRReisizeNormImgToBatchvalidate_text_rec_image_array TextRecResultz python-bidi) get_display) zPP-OCRv5_server_reczPP-OCRv5_mobile_receslav_PP-OCRv5_mobile_reckorean_PP-OCRv5_mobile_reclatin_PP-OCRv5_mobile_reczen_PP-OCRv5_mobile_recth_PP-OCRv5_mobile_recel_PP-OCRv5_mobile_recarabic_PP-OCRv5_mobile_recte_PP-OCRv5_mobile_recta_PP-OCRv5_mobile_recdevanagari_PP-OCRv5_mobile_reccyrillic_PP-OCRv5_mobile_reccCs|drtS|dvr tS|dvrtS|dvrtS|dkrtS|dkr%tS|dvr+tS|dkr1tS|d vr7t S|d vr=t S|d vrCt SdS) N)zPP-OCRz en_PP-OCR)zlatin_PP-OCRv3_mobile_recr&)zcyrillic_PP-OCRv3_mobile_recr-r$)zkorean_PP-OCRv3_mobile_recr%r'r(zarabic_PP-OCRv3_mobile_recr)zka_PP-OCRv3_mobile_rec)zte_PP-OCRv3_mobile_recr*)zta_PP-OCRv3_mobile_recr+)zdevanagari_PP-OCRv3_mobile_recr,) startswithrrr r rr rr rrr ) model_namer1t/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/models/text_recognition/predictor.pyget_text_rec_vis_fontDs. r3cseZdZiZeeZdddfdd ZddZdd Zd d Z dd d Z edddZ edddZ ddZ edddZedddZddZZS)TextRecRunnerPredictorNF) input_shapereturn_word_boxcs<tj|i|||_||_||_|\|_|_dSN) super__init__r5r6 get_vis_fontvis_font_buildpre_tfspost_op)selfr5r6argskwargs __class__r1r2r9}s  zTextRecRunnerPredictor.__init__cCtSr7rr?r1r1r2_build_batch_samplerz+TextRecRunnerPredictor._build_batch_samplercCtSr7r!rEr1r1r2_get_result_classz(TextRecRunnerPredictor._get_result_classc Csdtddi}|jddD]2}t|d}||jvsJ|j|}||i}|r4||fi|n||\}}|r@|||<qt|d<|jd i|jd}||fS) NReadRGBformat PreProcess transform_opsrr PostProcessr1)rconfiglistkeys _FUNC_MAPgetrbuild_postprocess) r?r=cfgZtf_keyfuncr@nameopr>r1r1r2r<s    zTextRecRunnerPredictor._buildcCs|jd|jd}t|D] \}}t||dq g}|D]}||jdt|jdqtt |}|jd|d}|jd|d} |j | d} |j j } t |} td d |jd d D} | dd \}}}||}t| | }g}td|D]}|||jdd\}}|d|}t||}||q{|j| |p|j||d\}}|jdvrdd|D}|j|j||||jgt |dS)NrKZimgsindexrr ReisizeNormr)xcss$|] }d|vr|ddVqdS) RecResizeImg image_shapeNr1).0r[r1r1r2 s z1TextRecRunnerPredictor.process..rOrPrrg?)r6 wh_ratio_list max_wh_ratior.cSs2g|]}t|trt|d|dfnt|qS)rr) isinstancetupler#)rcsr1r1r2 s$z2TextRecRunnerPredictor.process..Z input_pathZ page_indexZ input_imgZrec_textZ rec_scorer;)r= instances enumerater appendshapefloatnpZargsortarrayrunnerZ batch_samplerZ batch_sizelennextrRminrangemaxr>r6r0 input_paths page_indexesr;)r? batch_datar6batch_raw_imgsiimgZ width_listindicesZ batch_imgsr`Z batch_predsZ batch_numZimg_numrec_image_shapeZimgCZimgHZimgWrfZ end_img_noreinohwZwh_ratiotextsscoresr1r1r2processsR        zTextRecRunnerPredictor.processZ DecodeImagecCs|dksJdt|dfS)NFrKrMr)r?Z channel_firstZimg_moder1r1r2 build_readimgs z$TextRecRunnerPredictor.build_readimgracKsdt||jdfS)Nr_)rr5)rr5)r?rbrAr1r1r2 build_resizesz#TextRecRunnerPredictor.build_resizecKs$|ddkrt|ddSt)NrZrcharacter_dictcharacter_list)rVr Exception)r?rAr1r1r2rWs z(TextRecRunnerPredictor.build_postprocessZMultiLabelEncodecOdSN)NNr1r?r@rAr1r1r2foozTextRecRunnerPredictor.fooZKeepKeyscOrrr1rr1r1r2rrcCs t|jSr7)r3r0rEr1r1r2r:s z#TextRecRunnerPredictor.get_vis_font)F)__name__ __module__ __qualname__rUrregisterr9rFrIr<rrrrWrr: __classcell__r1r1rBr2r4ws$ 2    r4cspeZdZdZdddeffddZddZd d Zd d Zd dZ ddZ dde efddZ ddZ ZS)TextRecTransformersPredictorz?Text recognition predictor backed by Hugging Face transformers.F)r6r6csNtj|i|||_t|j|_tdd|_|\|_ |_ | |_ dS)NrLrM) r8r9r6r3r0r;rread_opr<image_processorZinfer_build_postprocessr>)r?r6r@rArBr1r2r9s   z%TextRecTransformersPredictor.__init__cCrDr7rrEr1r1r2rFrGz1TextRecTransformersPredictor._build_batch_samplercCrHr7r!rEr1r1r2rIrJz.TextRecTransformersPredictor._get_result_classcCs,ddlm}m}||}||}||fS)Nr)AutoImageProcessorAutoModelForTextRecognition)Z transformersrrZ_load_pretrained_processorZ_load_pretrained_model)r?rrrmodelr1r1r2r<s  z#TextRecTransformersPredictor._buildcCsbt|jdd}|s|jdid}|s tt|jjd|ddkr,|dd}t|dS) NrrQrzR does not provide the character dictionary required for text recognition decoding.rblankrr)getattrrZ model_configrV RuntimeErrortyperr)r?rr1r1r2rs   z/TextRecTransformersPredictor._build_postprocesscCs^t|jdipi}t|jdip|}t|d|dd}t|d|dd}d||fS)Nsizepad_sizeheight0widthi@r)rrintrV)r?rrZimg_hZimg_wr1r1r2_get_rec_image_shapes  z1TextRecTransformersPredictor._get_rec_image_shapeNc Cs|rtd|j|jd}t|D] \}}t||dqdd|D}|j|d}||}||\} } |j |j || | |j gt |dS)Nz5transformers engine doesn't support `return_word_box`r\r]cSsg|]}t|qSr1)rZ fromarray)rcr~r1r1r2rjsz8TextRecTransformersPredictor.process..)imagesrk) rwarningrrlrmr Zpreprocess_imagesforward postprocessryrzr;rt) r?r{r6r|r}r~rZ model_inputsoutputsrrr1r1r2rs    z$TextRecTransformersPredictor.processcKs0|j|}dd|D}dd|D}||fS)NcSg|]}|dqS)textr1rcrr1r1r2rj0z.cSr)Zscorer1rr1r1r2rj1r)rZpost_process_text_recognition)r?rrAresultsrrr1r1r2r.s z(TextRecTransformersPredictor.postprocessr7)rrr__doc__boolr9rFrIr<rrrrrrr1r1rBr2rsr)-typingrnumpyrqZPILrutilsrZ utils.depsrrZ utils.fontsrr r r r r rrrrrZutils.func_registerrZcommon.batch_samplerrZ common.readerrZ predictorsrrZ processorsrrrr resultr"Zbidi.algorithmr#ZTEXT_REC_TRANSFORMERS_MODELSr3r4rr1r1r1r2s&   4    3r