o 'j@sddlZddlZddlZejeZddlZejej edddl Z ddl Z ddl Z ddlmZddlZddlmZddlmZdddZed ej ed d d Zed dZeddZddlmZddlmZddlmZm Z m!Z!m"Z"ddl#m$Z$m%Z%m&Z&m'Z'ddl(m)Z)m*Z*m+Z+ddl,m-Z-m.Z.ddl/m0Z0m1Z1m2Z2eZ3gdZ4dgZ5dZ6ddgZ7ej8dZ9dZ:gdZ;dZd(d? d@d!dAiidBd!dCid!d#id!d$id%dDd'd(dEd*d(dFd,d(dGd.d(d/d0d(dHd2d(dId4d(dJd6d(d7d8d(dKd:d(d;dd(d? d@d!dAiidBd@d!dMiid@dNd'd(id@d!dAiidBd!dOid!dPid!dQidRdSd'd(dTd*d(dUdVd(dWdXd(dYd,d(dZd.d(d[d0d(d\d2d(d]d4d(d^d6d(d_d8d(d`d:d(dadd(dcddd(ded@d!dAiidBdfdgdhdidjd(iidkdjd(dldmd(dndodpd(dqdrd(dndsdtduZ>ddvdwZ?dxdyZ@dzd{ZAd|eBfd}d~ZCddZDGdddejEZFGddde0ZGddZHdS)N)Path)BytesIO)ImageFcCs8tj||}tj|}|j||r|tj|<|SN) importlibutilspec_from_file_locationmodule_from_specloader exec_modulesysmodules) module_name file_pathmake_importablespecmodulerT/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddleocr/paddleocr.py _import_file"s    rtoolsztools/__init__.pyT)rppocrZ paddleocr ppstructure) get_logger)predict_system)check_and_readget_image_file_listalpha_to_color binarize_img)maybe_downloaddownload_with_progressbaris_linkconfirm_model_dir_url)draw_ocrstr2bool check_gpu) init_argsdraw_structure_result)StructureSystemsave_structure_resto_excel) PaddleOCR PPStructurer$r(r*r!r+ZDBz2.7.3ZCRNNZ SVTR_LCNetz ~/.paddleocr/PP-OCRv4)PP-OCRPP-OCRv2PP-OCRv3r.PP-StructureV2 PP-StructureurlzJhttps://paddleocr.bj.bcebos.com/PP-OCRv4/chinese/ch_PP-OCRv4_det_infer.tarzJhttps://paddleocr.bj.bcebos.com/PP-OCRv3/english/en_PP-OCRv3_det_infer.tarzYhttps://paddleocr.bj.bcebos.com/PP-OCRv3/multilingual/Multilingual_PP-OCRv3_det_infer.tar)chenmlzJhttps://paddleocr.bj.bcebos.com/PP-OCRv4/chinese/ch_PP-OCRv4_rec_infer.tarz./ppocr/utils/ppocr_keys_v1.txt)r4 dict_pathzJhttps://paddleocr.bj.bcebos.com/PP-OCRv4/english/en_PP-OCRv4_rec_infer.tarz./ppocr/utils/en_dict.txtzShttps://paddleocr.bj.bcebos.com/PP-OCRv4/multilingual/korean_PP-OCRv4_rec_infer.tarz"./ppocr/utils/dict/korean_dict.txtzRhttps://paddleocr.bj.bcebos.com/PP-OCRv4/multilingual/japan_PP-OCRv4_rec_infer.tarz!./ppocr/utils/dict/japan_dict.txtzXhttps://paddleocr.bj.bcebos.com/PP-OCRv3/multilingual/chinese_cht_PP-OCRv3_rec_infer.tarz'./ppocr/utils/dict/chinese_cht_dict.txtzOhttps://paddleocr.bj.bcebos.com/PP-OCRv4/multilingual/ta_PP-OCRv4_rec_infer.tarz./ppocr/utils/dict/ta_dict.txtzOhttps://paddleocr.bj.bcebos.com/PP-OCRv4/multilingual/te_PP-OCRv4_rec_infer.tarz./ppocr/utils/dict/te_dict.txtzOhttps://paddleocr.bj.bcebos.com/PP-OCRv4/multilingual/ka_PP-OCRv4_rec_infer.tarz./ppocr/utils/dict/ka_dict.txtzRhttps://paddleocr.bj.bcebos.com/PP-OCRv3/multilingual/latin_PP-OCRv3_rec_infer.tarz!./ppocr/utils/dict/latin_dict.txtzShttps://paddleocr.bj.bcebos.com/PP-OCRv4/multilingual/arabic_PP-OCRv4_rec_infer.tarz"./ppocr/utils/dict/arabic_dict.txtzUhttps://paddleocr.bj.bcebos.com/PP-OCRv3/multilingual/cyrillic_PP-OCRv3_rec_infer.tarz$./ppocr/utils/dict/cyrillic_dict.txtzWhttps://paddleocr.bj.bcebos.com/PP-OCRv4/multilingual/devanagari_PP-OCRv4_rec_infer.tarz&./ppocr/utils/dict/devanagari_dict.txt) r5r6koreanjapan chinese_chttatekalatinarabiccyrillic devanagarir5zRhttps://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_mobile_v2.0_cls_infer.tar)detrecclszJhttps://paddleocr.bj.bcebos.com/PP-OCRv3/chinese/ch_PP-OCRv3_det_infer.tarzJhttps://paddleocr.bj.bcebos.com/PP-OCRv3/chinese/ch_PP-OCRv3_rec_infer.tarzJhttps://paddleocr.bj.bcebos.com/PP-OCRv3/english/en_PP-OCRv3_rec_infer.tarzShttps://paddleocr.bj.bcebos.com/PP-OCRv3/multilingual/korean_PP-OCRv3_rec_infer.tarzRhttps://paddleocr.bj.bcebos.com/PP-OCRv3/multilingual/japan_PP-OCRv3_rec_infer.tarzOhttps://paddleocr.bj.bcebos.com/PP-OCRv3/multilingual/ta_PP-OCRv3_rec_infer.tarzOhttps://paddleocr.bj.bcebos.com/PP-OCRv3/multilingual/te_PP-OCRv3_rec_infer.tarzOhttps://paddleocr.bj.bcebos.com/PP-OCRv3/multilingual/ka_PP-OCRv3_rec_infer.tarzShttps://paddleocr.bj.bcebos.com/PP-OCRv3/multilingual/arabic_PP-OCRv3_rec_infer.tarzWhttps://paddleocr.bj.bcebos.com/PP-OCRv3/multilingual/devanagari_PP-OCRv3_rec_infer.tarzJhttps://paddleocr.bj.bcebos.com/PP-OCRv2/chinese/ch_PP-OCRv2_det_infer.tarzJhttps://paddleocr.bj.bcebos.com/PP-OCRv2/chinese/ch_PP-OCRv2_rec_infer.tarzRhttps://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_mobile_v2.0_det_infer.tarz\https://paddleocr.bj.bcebos.com/dygraph_v2.0/multilingual/en_ppocr_mobile_v2.0_det_infer.tarz[https://paddleocr.bj.bcebos.com/dygraph_v2.0/table/en_ppocr_mobile_v2.0_table_det_infer.tar)r5r6 structurezRhttps://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_mobile_v2.0_rec_infer.tarz]https://paddleocr.bj.bcebos.com/dygraph_v2.0/multilingual/en_number_mobile_v2.0_rec_infer.tarzZhttps://paddleocr.bj.bcebos.com/dygraph_v2.0/multilingual/french_mobile_v2.0_rec_infer.tarz"./ppocr/utils/dict/french_dict.txtzZhttps://paddleocr.bj.bcebos.com/dygraph_v2.0/multilingual/german_mobile_v2.0_rec_infer.tarz"./ppocr/utils/dict/german_dict.txtzZhttps://paddleocr.bj.bcebos.com/dygraph_v2.0/multilingual/korean_mobile_v2.0_rec_infer.tarzYhttps://paddleocr.bj.bcebos.com/dygraph_v2.0/multilingual/japan_mobile_v2.0_rec_infer.tarz_https://paddleocr.bj.bcebos.com/dygraph_v2.0/multilingual/chinese_cht_mobile_v2.0_rec_infer.tarzVhttps://paddleocr.bj.bcebos.com/dygraph_v2.0/multilingual/ta_mobile_v2.0_rec_infer.tarzVhttps://paddleocr.bj.bcebos.com/dygraph_v2.0/multilingual/te_mobile_v2.0_rec_infer.tarzVhttps://paddleocr.bj.bcebos.com/dygraph_v2.0/multilingual/ka_mobile_v2.0_rec_infer.tarz_https://paddleocr.bj.bcebos.com/dygraph_v2.0/multilingual/latin_ppocr_mobile_v2.0_rec_infer.tarz`https://paddleocr.bj.bcebos.com/dygraph_v2.0/multilingual/arabic_ppocr_mobile_v2.0_rec_infer.tarzbhttps://paddleocr.bj.bcebos.com/dygraph_v2.0/multilingual/cyrillic_ppocr_mobile_v2.0_rec_infer.tarzdhttps://paddleocr.bj.bcebos.com/dygraph_v2.0/multilingual/devanagari_ppocr_mobile_v2.0_rec_infer.tarz[https://paddleocr.bj.bcebos.com/dygraph_v2.0/table/en_ppocr_mobile_v2.0_table_rec_infer.tarzppocr/utils/dict/table_dict.txt)r5r6frenchgermanr9r:r;r<r=r>r?r@rArBrF)r.r1r0r/tabler6zahttps://paddleocr.bj.bcebos.com/dygraph_v2.0/table/en_ppocr_mobile_v2.0_table_structure_infer.tarz)ppocr/utils/dict/table_structure_dict.txtzehttps://paddleocr.bj.bcebos.com/ppstructure/models/slanet/en_ppstructure_mobile_v2.0_SLANet_infer.tarzehttps://paddleocr.bj.bcebos.com/ppstructure/models/slanet/ch_ppstructure_mobile_v2.0_SLANet_infer.tarz,ppocr/utils/dict/table_structure_dict_ch.txt)r6r5zahttps://paddleocr.bj.bcebos.com/ppstructure/models/layout/picodet_lcnet_x1_0_fgd_layout_infer.tarz6ppocr/utils/dict/layout_dict/layout_publaynet_dict.txtzfhttps://paddleocr.bj.bcebos.com/ppstructure/models/layout/picodet_lcnet_x1_0_fgd_layout_cdla_infer.tarz1ppocr/utils/dict/layout_dict/layout_cdla_dict.txt)rIlayout)r3r2)OCR STRUCTUREcCsddl}t}||_|jdtdd|jdtdd|jdtdd|jdtd d|jd ttd d d |jdttddd |jD] }|j dvrKd|_ qA|rR| Si}|jD]}|j ||j <qW|j di|S)Nrz--langr5)typedefaultz--detTz--recz--typeocrz --ocr_versionr.aUOCR Model version, the current model support list is as follows: 1. PP-OCRv4/v3 Support Chinese and English detection and recognition model, and direction classifier model2. PP-OCRv2 Support Chinese detection and recognition model. 3. PP-OCR support Chinese detection, recognition and direction classifier and multilingual recognition model.)rMchoicesrNhelpz--structure_versionr2zModel version, the current model support list is as follows: 1. PP-Structure Support en table structure model. 2. PP-StructureV2 Support ch and en table structure model.)rec_char_dict_pathtable_char_dict_pathlayout_dict_pathr) argparser'add_help add_argumentstrr%SUPPORT_OCR_MODEL_VERSIONSUPPORT_STRUCTURE_MODEL_VERSION_actionsdestrN parse_args Namespace)mMainrUparseractionZinference_args_dictrrrr]s>   r]cCsgd}gd}gd}gd}||vrd}n||vrd}n ||vr%d}n||vr+d}|td td vsEJd td td ||d krOd }||fS|d krYd }||fS|dvrcd}||fSd}||fS)N)*afazbscscydadeesetfrgahrhuidisitZkuZlaltlvmimsmtnlnoocpiplptroZrs_latinskslsqsvswtltruzvirGrH)arfaZugur)ruZ rs_cyrillicbebgukZmnZabqZadyZkbdZavaZdarZinhZcheZlbeZleztab) himrneZbhmaiangZbhoZmahZscknewZgomsaZbgcr?r@rArBrKrDz!param lang must in {}, but got {}r5rF)r6r?r6r7) MODEL_URLSDEFAULT_OCR_MODEL_VERSIONformatkeys)langZ latin_langZ arabic_langZ cyrillic_langZdevanagari_langdet_langrrr parse_langs< rcCs|dkrt}n |dkrt}ntt|}||vr|}|||vr;|||vr)|}ntd|||t d||||vrc||||vrN|}ntd|||||t d||||S)NrKrLz,{} models is not support, we only support {}z8lang {} is not support, we only support {} for {} models) rDEFAULT_STRUCTURE_MODEL_VERSIONNotImplementedErrorrloggererrorrrr exit)rMversionZ model_typerZDEFAULT_MODEL_VERSIONZ model_urlsrrrget_model_configs4     rcontentcCstj|tjd}t|tjS)N)Zdtype)np frombufferuint8cv2imdecodeZIMREAD_UNCHANGED)rZnp_arrrrr img_decodesrcCspt|tr t|}t|trt|rt|dd}|}t|\}}}|s|st|d}|}t|}Wdn1s>wY|durz>t }t |}t |}| d} | |d| d|} tt| dd} t| } t| tj} t| tj}Wntd|YdS|durtd|dSt|tjrt|jd krt|tj}|S) Ntmp.jpgrbRGBZjpegrzutf-8)encodingerror in loading image:{}) isinstancebytesrrXr"r!ropenreadrrconvertsaveseekbase64 b64encode b64decoderrrrrZ IMREAD_COLORrrrndarraylenshapeZcvtColorZCOLOR_GRAY2BGR)imgZ image_fileflag_gifflag_pdffZimg_strbufimageZimrgbZ image_bytesZ data_base64Z image_decodeZ img_arrayrrr check_imgsJ           rcs2eZdZfddZ      dddZZS) r,c  stdd}|jjdi||jtvsJdt|jt|j|_|js+t t j |j |_ t|j\}}td|jd|}t|jtjtdd||d\|_}td|jd|}t|jtjtdd||d\|_}td|jd d } t|jtjtdd | d\|_} |jd vrd |_nd |_|jst|j|t|j|t|j| |jtvrt dtt d|j!t"vrt dt"t d|j#durt$t%t&j'|d|_#t (|t)*||j+|_+dS)zm paddleocr package args: **kwargs: other params show in paddleocr --help Fr_z"ocr_version must in {}, but get {}rKrCwhlr4rDrEr5)r1r.z 3, 48, 320z 3, 32, 320zdet_algorithm must in {}rzrec_algorithm must in {}Nr8r),r]__dict__update ocr_versionrYrr&use_gpushow_logrsetLevelloggingINFO use_angle_clsrrrr# det_model_dirospathjoinBASE_DIR rec_model_dirZ cls_model_dirZrec_image_shapeZuse_onnxr Z det_algorithmSUPPORT_DET_MODELrr rZ rec_algorithmSUPPORT_REC_MODELrRrXr__file__parentdebugsuper__init__page_num) selfkwargsparamsrrdet_model_configdet_urlrec_model_configrec_urlZcls_model_configZcls_url __class__rrr/sh                  zPaddleOCR.__init__TFrrcst|tjtttfs Jt|tr|dkrtdtd|dkr,|j dkr,t dt |}t|trN|j t |ksA|j dkrFt ||_ |d|j }n|g}fdd} |r|rg} t|D]*\} }| |}|||\} } }| s~| s~| dqcd d t| | D}| |qc| S|r|sg} t|D]#\} }| |}||\} }| s| dqd d | D}| |q| Sg} g}t|D]0\} }t|ts| |}|g}|j r|r||\}}}|s||||\} }| | q|s|S| S) u> OCR with PaddleOCR args： img: img for OCR, support ndarray, img_path and list or ndarray det: use text detection or not. If False, only rec will be exec. Default is True rec: use text recognition or not. If False, only det will be exec. Default is True cls: use angle classifier or not. Default is True. If True, the text with rotation of 180 degrees can be recognized. If no text is rotated by 180 degrees, use cls=False to get better performance. Text with rotation of 90 or 270 degrees can be recognized even if cls=False. bin: binarize image to black and white. Default is False. inv: invert image colors. Default is False. alpha_color: set RGB color Tuple for transparent parts replacement. Default is pure white. Tz.When input a list of images, det must be falserFz]Since the angle classifier is not initialized, it will not be used during the forward processNcs(t|}r t|}rt|}|Sr)rrZ bitwise_notr)Z_image alpha_colorbininvrrpreprocess_images  z'PaddleOCR.ocr..preprocess_imagecSsg|] \}}||gqSrtolist).0boxresrrr sz!PaddleOCR.ocr..cSsg|]}|qSrr)rrrrrrs)rrrlistrXrrrrrwarningrrr enumerate__call__appendzipZ text_detectorZtext_classifierZtext_recognizer)rrrCrDrErrrZimgsrZocr_residxZdt_boxesZrec_res_Ztmp_resZelapseZcls_resZ cls_res_tmprrrrOksl           z PaddleOCR.ocr)TTTFFr)__name__ __module__ __qualname__rrO __classcell__rrrrr,.s >r,cs*eZdZfddZdfdd ZZS)r-c stdd}|jjdi||jtvsJdt|jt|j|_d|_|j s.t t j t|j\}}|dkrs            ?W?VN&  O)&( B