o e5v@s:ddlZddlZddlZejeZddlZejej edddl Z ddl Z ddl Z ddlmZeddZeddZeddZddlmZdd lmZeZdd lmZmZdd lmZmZmZm Z dd l!m"Z"m#Z#m$Z$dd l%m&Z&m'Z'ddl(m)Z)m*Z*m+Z+gdZ,dgZ-dZ.ddgZ/ej0dZ1dZ2gdZ3dZ4ddgZ5ddiddiddidddd d!d"d d#d$d d%d&d d'd(d d)d*d d+d,d d-d.d d/d0d d1d2d d3d4d d5d6d d7 d8dd9iid:d8dd;iid8didd?id@dAdd dBd"d dCdDd dEdFd dGd$d dHd&d dId(d dJd*d dKd,d dLd.d dMd0d dNd2d dOd4d dPd6d dQdRd dSd8dd9iid:dTdUdVdWdXd iidYdXd dZd[d d\d]d^d d_d`d d\dadbdcZ6dvdedfZ7dgdhZ8didjZ9dke:fdldmZ;dndoZGdrdsdse)Z?dtduZ@dS)wN)Path.toolsppocr ppstructure)predict_system) get_logger)check_and_readget_image_file_list)maybe_downloaddownload_with_progressbaris_linkconfirm_model_dir_url)draw_ocrstr2bool check_gpu) init_argsdraw_structure_result)StructureSystemsave_structure_resto_excel) PaddleOCR PPStructurerrrr rZDBz2.6.1.1ZCRNNZ SVTR_LCNetz ~/.paddleocr/PP-OCRv3)PP-OCRPP-OCRv2rPP-StructureV2 PP-StructureurlzJhttps://paddleocr.bj.bcebos.com/PP-OCRv3/chinese/ch_PP-OCRv3_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-OCRv3/chinese/ch_PP-OCRv3_rec_infer.tarz./ppocr/utils/ppocr_keys_v1.txt)r dict_pathzJhttps://paddleocr.bj.bcebos.com/PP-OCRv3/english/en_PP-OCRv3_rec_infer.tarz./ppocr/utils/en_dict.txtzShttps://paddleocr.bj.bcebos.com/PP-OCRv3/multilingual/korean_PP-OCRv3_rec_infer.tarz"./ppocr/utils/dict/korean_dict.txtzRhttps://paddleocr.bj.bcebos.com/PP-OCRv3/multilingual/japan_PP-OCRv3_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-OCRv3/multilingual/ta_PP-OCRv3_rec_infer.tarz./ppocr/utils/dict/ta_dict.txtzOhttps://paddleocr.bj.bcebos.com/PP-OCRv3/multilingual/te_PP-OCRv3_rec_infer.tarz./ppocr/utils/dict/te_dict.txtzOhttps://paddleocr.bj.bcebos.com/PP-OCRv3/multilingual/ka_PP-OCRv3_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-OCRv3/multilingual/arabic_PP-OCRv3_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-OCRv3/multilingual/devanagari_PP-OCRv3_rec_infer.tarz&./ppocr/utils/dict/devanagari_dict.txt) r r!koreanjapan chinese_chttatekalatinarabiccyrillic devanagarir zRhttps://paddleocr.bj.bcebos.com/dygraph_v2.0/ch/ch_ppocr_mobile_v2.0_cls_infer.tardetrecclszJhttps://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)r r! 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)r r!frenchgermanr$r%r&r'r(r)r*r+r,r-r2)rrrtabler!zahttps://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)r!r zahttps://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)r5layout)rr)OCR STRUCTURETcCsddl}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--langr )typedefaultz--detTz--recz--typeocrz --ocr_versionraROCR Model version, the current model support list is as follows: 1. PP-OCRv3 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.)r9choicesr:helpz--structure_versionrzModel 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_path) argparseradd_help add_argumentstrrSUPPORT_OCR_MODEL_VERSIONSUPPORT_STRUCTURE_MODEL_VERSIONZ_actionsdestr: parse_args Namespace)mMainrBparseractionZinference_args_dictrArACD:\Projects\ConvertPro\env\Lib\site-packages\paddleocr/paddleocr.pyrI3s>   rIcCsgd}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_latinskslsqsvswtltruzvir3r4)arfaZugur)ruZ rs_cyrillicbebgukZmnZabqZadyZkbdZavaZdarZinhZcheZlbeZleztab) himrneZbhmaiangZbhoZmahZscknewZgomsaZbgcr*r+r,r-r7r0z!param lang must in {}, but got {}r r2)r!r*r!r") MODEL_URLSDEFAULT_OCR_MODEL_VERSIONformatkeys)langZ latin_langZ arabic_langZ cyrillic_langZdevanagari_langdet_langrArArN parse_lang\s< rcCs|dkrt}n |dkrt}ntt|}||vr|}|||vr;|||vr)|}ntd|||t d||||vrc||||vrN|}ntd|||||t d||||S)Nr7r8z,{} models is not support, we only support {}z8lang {} is not support, we only support {} for {} models) rDEFAULT_STRUCTURE_MODEL_VERSIONNotImplementedErrorrloggererrorrrsysexit)r9versionZ model_typerZDEFAULT_MODEL_VERSIONZ model_urlsrArArNget_model_configs4     rcontentcCstj|tjd}t|tjS)N)Zdtype)npZ frombufferZuint8cv2ZimdecodeZ IMREAD_COLOR)rZnp_arrrArArN img_decodesrcCst|tr t|}t|trOt|rt|dd}|}t|\}}}|sA|sAt|d}t|}Wdn1srEr__file__parentdebugsuper__init__page_num) selfkwargsparamsrrdet_model_configdet_urlrec_model_configrec_urlZcls_model_configZcls_url __class__rArNrsh                  zPaddleOCR.__init__TcCst|tjtttfs Jt|tr|dkrtdtd|dkr,|j dkr,t dt |}t|trN|j t |ksA|j dkrFt ||_ |d|j }n|g}|rz|rzg}t|D]\}}|||\}} } ddt|| D} || q[|S|r|sg}t|D]\}}||\}} d d|D} || q|Sg}g} t|D],\}}t|ts|g}|j r|r||\}}} |s| |||\} } || q|s| S|S) uY 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. Tz.When input a list of images, det must be falserFzoSince the angle classifier is not initialized, the angle classifier will not be uesd during the forward processNcSsg|] \}}||gqSrAtolist).0boxresrArArN sz!PaddleOCR.ocr..cSsg|]}|qSrAr)rrrArArNrs)rrrlistrErrrrrwarningrrr enumerate__call__zipappendZ text_detectorZtext_classifierZtext_recognizer)rrr/r0r1ZimgsZocr_residxZdt_boxesZrec_res_Ztmp_resZelapseZcls_resZ cls_res_tmprArArNr;sX          z PaddleOCR.ocr)TTT)__name__ __module__ __qualname__rr; __classcell__rArArrNrs rErrrr?r@rrr)rrrrrZ table_langrrrrZtable_model_configZ table_urlZlayout_model_configZ layout_urlrrArNr4s|                  zPPStructure.__init__Frcs"t|}tj|||d\}}|S)Nimg_idx)rrr)rrZreturn_ocr_result_in_tablerrrrrArNrns  zPPStructure.__call__)Fr)rrrrrrrArArrNr3s :rc! Cstdd}|j}t|rt|ddg}nt|j}t|dkr+td|jdS|j dkr9t di|j }n|j dkrGt di|j }nt |D]}tj|dd}td d |d |j dkr|j||j|j|jd }|durtt|D]}||}|D]} t| qq~qK|j dkrt|\} } } | s| st|} |jr|jr| rdd lm} tj|j d |}| |}|!||"td|qK| s| durtd|qK|| gg}n6g}t#| D]/\}}tj$tj|j |ddtj|j ||dt%|d}t&|||'||gqg}t#|D]U\}\}} td|dt|tj|dd}|||d}t(||j |||jrz|gkrzddl)m*}ddl+m,}| j-\}}}||}|||}||7}q&|jr|gkrzddl+m.}|| ||j |Wnt/y}ztd||WYd}~qKd}~ww|D]} | 0d| 0dt| qtd|j qKdS)NTrrrzno images find in {}r;r2rz{}{}{}z **********r.) Converterz{}.docxzdocx save to {}r)exist_okrz.jpgzprocessing {}/{} page:r)deepcopy)sorted_layout_boxes)convert_info_docxz.error in layout recovery image:{}, err msg: {}rrzresult save to {}rA)1rI image_dirrr r rrrrr9rrrrrrbasenamesplitinfor;r/r0rranger rZimreadZrecoveryZuse_pdf2docx_apiZpdf2docx.converterrroutputconvertclosermakedirsrEZimwriterrcopyrZ$ppstructure.recovery.recovery_to_docrrr Exceptionpop)!argsrZimage_file_listZengineZimg_pathZimg_nameresultrrlinerrrrZ docx_fileZcvZ img_pathsindexZpdf_imgZ pdf_img_pathZall_resZ new_img_pathZ new_img_namerrhwrZ result_cpZ result_sortedrexitemrArArNmainus                         r)T)Arr importlibrdirnamer__dir__Zpaddlerrrrnumpyrpathlibr import_modulerrrZ tools.inferrZppocr.utils.loggingr rZppocr.utils.utilityr r Zppocr.utils.networkr r rrZtools.infer.utilityrrrZppstructure.utilityrrZppstructure.predict_systemrrr__all__rVERSIONr expanduserrrrFrrGrrIrrrrrZ TextSystemrrrrArArArNs        ?VNQ  z)&w B