o 'j\@s`ddlZddlZddlmZmZmZddlZddlm Z m Z m Z m Z ddZ ddZdd ZdS) N)Image ImageDraw ImageFont)draw_ocr_box_txtstr2bool str2int_tuple init_argscCst}|jdtdd|jdtdd|jdtdd|jdtd |jd td d|jd td d|jdtd |jdtdd|jdtddd|jdtddd|jdtdd|jdtd |jdtd |jdtd d|jdtdd|jdtdd|jdtd d!gd d"d#|jd$td%d&d|jd'td d(d|jd)td d*d|jd+td d,d|jd-td%d.d|jd/td%d0d|jd1td%d2d|jd3td%d4d|jd5td6d7d|S)8Nz--outputz./output)typedefaultz--table_max_leniz--table_algorithmZ TableAttnz--table_model_dir)r z--merge_no_span_structureTz--table_char_dict_pathz/../ppocr/utils/dict/table_structure_dict_ch.txtz--layout_model_dirz--layout_dict_pathz9../ppocr/utils/dict/layout_dict/layout_publaynet_dict.txtz--layout_score_thresholdg?zThreshold of score.)r r helpz--layout_nms_thresholdzThreshold of nms.z--kie_algorithmZ LayoutXLMz--ser_model_dirz--re_model_dirz--use_visual_backbonez--ser_dict_pathz'../train_data/XFUND/class_list_xfun.txtz--ocr_order_methodz--modeZ structureZkiezstructure and kie is supported)r choicesr r z--image_orientationFz/Whether to enable image orientation recognitionz--layoutz!Whether to enable layout analysisz--tablez=In the forward, whether the table area uses table recognitionz--ocrz@In the forward, whether the non-table area is recognition by ocrz --recoveryz$Whether to enable layout of recoveryz--use_pdf2docx_apizWhether to use pdf2docx apiz--invertz)Whether to invert image before processingz --binarizez5Whether to threshold binarize image before processingz --alphacolorrrzQReplacement color for the alpha channel, if the latter is present; R,G,B integers) infer_args add_argumentstrintrfloatboolrparserr^/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddleocr/ppstructure/utility.pyrsrcCst}|S)N)r parse_argsrrrrrxsrcCst|tjr t|}ggg}}}|}t|}d}d} i} d} tj || dd} |D]} | d| vrNt ddt ddt ddf}|| | d<n| | d}| d }|j |d|d f|d |d fg|d d | | d\}}}}||||}}|j |d|d f|d||d |fg| d|j|d|d f| d|| d| ddkrq.| dD]}|t|d||d||dqq.t|||||dd}|S)Nr )Przutf-8)encodingr rrZbbox)outlinewidth)fill)r#fonttableresZ text_regiontextZ confidence) font_pathZ drop_score) isinstancenpZndarrayrZ fromarraycopyrZDrawrZtruetyperandomrandintZ rectangleZgetbboxr'appendarrayr)imageresultr(ZboxesZtxtsZscoresZ img_layoutZ draw_layoutZ text_colorZtext_background_colorZ catid2colorZ font_sizer$regionZ box_colorZ box_layoutlefttoprightbottomZtext_wZtext_hZ text_resultZim_showrrrdraw_structure_result}s`         r7)r,astZPILrrrnumpyr*Ztools.infer.utilityrrrrrrr7rrrrs c