o ex6 @sddlZddlZejejeZejeejdejej eddej d<ddl Z ddl Z ddlZddlZddlmmZddlmZddlmZmZddlmZmZddlmZddlZeZGd d d eZe d kre!Z"ee"j#Z$ee"Z%dZ&e"j'Z'ej(e'd d e"j)re j*+ddgd,e j-Z.e/dD]Z0e%e.Z1qgZ2e3e$D]\Z4Z5ee5\Z.Z6Z7e6se7se 8e5Z.e7se.dure9d:e5qe.gZ;ne"jZ.eZ?e%e.\Z@ZAee?ZBe&eB7Z&e=e;dkr/ejCe5deDe>deDeEdde@DdZFnejCe5deDeEdde@DdZFe2eFeGeFe=e;dkrbeGd:e4e>e5eBn eGd:e4e5eBeHe@e.ZIe6r~e5dddZJne7re5KddeDe>dZJne5ZJej e'd:ejCeJZLe MeLeIeGd :eLqqeNej e'd!d"ZOeOPe2eOQWdn 1swYe"jRre%jSTdSdSdS)#Nz../..Z auto_growthZFLAGS_allocator_strategy) get_logger)get_image_file_listcheck_and_read)create_operators transform)build_post_processc@s<eZdZddZddZddZddZd d Zd d Zd S) TextDetectorc Cs||_|j|_|j|_d|j|jdidgdgddddid did d d d giig}i}|jdkrZd|d<|j|d<|j|d<d|d<|j|d<|j|d<|j |d<|j |d<n|jdkrd|d<|j|d<|j|d<d|d<|j|d<|j|d<|j |d<|j |d<dgdgddddi|d<n|jdkrd|d<|j |d<|j |d <|j |d!<n|jd"krdd#|jii|d$<d%|d<|j|d<|j|d!<|j d&krd'|d(<d)|d*<d+|d,<nd-|d(<d.|d*<d/|d,<ns|jd0kr d1|d<|j|d<|j|d<|j|d2<|j |d<|j|d3<nO|jd4kr;dd5d6d7gii|d$<d8|d<|j|d9<|j|d:<|j|d;<|j|d<<|j |d<n!|jd=krNd>d?d@ii|d$<dA|d<ntdB|jtd$t||_t||_ t!"|dCt\|_#|_$|_%|_&|jr|j$j'd-d\}}|dur|dur|d$kr|d$krddD||gii|d$<t||_|j(rd$dl)}t*+}t!,}|j-dC|j.ddEd|j&|d|j/r|ndgdFd-tdG |_0dSdS)HNZDetResizeForTest)Zlimit_side_lenZ limit_typeZNormalizeImage)gZd;O?gy&1?g?)g ףp= ?gv/?gCl?z1./255.Zhwc)ZstdmeanscaleorderZ ToCHWImageZKeepKeysZ keep_keysimageshapeDBZ DBPostProcessnameZthreshZ box_threshiZmax_candidatesZ unclip_ratio use_dilationZ score_modeZbox_typeDB++)?rr)gy=?gy+H?gw? ?EASTZEASTPostProcessZ score_threshZ cover_threshZ nms_threshSASTZ resize_longrZSASTPostProcesspolyZsample_pts_numg333333?Z expand_scaleg?Zshrink_ratio_of_widthrg333333?PSEZPSEPostProcessZmin_arear FCEZ rescale_imgi8iZFCEPostProcessscalesalphabetafourier_degreeCTZScaleAlignedShortZ short_sizeZ CTPostProcesszunknown det_algorithm:{}Zdet image_shapeZdynamic)Zpreprocess_timeZinference_timeZpostprocess_time) Z model_nameZmodel_precisionZ batch_sizeZ data_shapeZ save_pathZinference_configZpidsZ process_nameZgpu_idsZ time_keyswarmuplogger)1args det_algorithmuse_onnxZdet_limit_side_lenZdet_limit_typeZ det_db_threshZdet_db_box_threshZdet_db_unclip_ratiorZdet_db_score_mode det_box_typeZdet_east_score_threshZdet_east_cover_threshZdet_east_nms_threshZdet_sast_score_threshZdet_sast_nms_threshZdet_pse_threshZdet_pse_box_threshZdet_pse_min_areaZ det_pse_scalerrrrr#infoformatsysexitr preprocess_oprpostprocess_oputilityZcreate_predictor predictor input_tensoroutput_tensorsconfigr benchmarkauto_logosgetpidZget_infer_gpuidZ AutoLogger precisionZuse_gpuautolog) selfr$Zpre_process_listZpostprocess_paramsZimg_hZimg_wr4pidZgpu_idr;QD:\Projects\ConvertPro\env\Lib\site-packages\paddleocr/tools/infer/predict_det.py__init__&s                                        (  zTextDetector.__init__cCstjddd}|jdd}|t||d<|t||d<tj|t|t|fdd}tjt|dd}|t||d<|t||d<|S) N)rZfloat32)ZdtyperZaxisrr)npZzerossumZargminZargmaxdeletediffarray)r9ZptsrectstmprDr;r;r<order_points_clockwises  z#TextDetector.order_points_clockwisecCsht|jdD]*}ttt||dfd|d||df<ttt||dfd|d||df<q|S)Nrr)ranger intminmax)r9points img_height img_widthZpnor;r;r< clip_det_ress(*zTextDetector.clip_det_resc Cs|dd\}}g}|D]C}t|turt|}||}||||}ttj|d|d}ttj|d|d}|dksI|dkrJq | |q t|}|S)Nrrrr@) typelistrArErIrQrKZlinalgZnormappend) r9dt_boxesr!rOrP dt_boxes_newboxZ rect_widthZ rect_heightr;r;r<filter_tag_det_ress     zTextDetector.filter_tag_det_rescCsZ|dd\}}g}|D]}t|turt|}||||}||q t|}|S)Nrr)rRrSrArErQrT)r9rUr!rOrPrVrWr;r;r<filter_tag_det_res_only_clips    z)TextDetector.filter_tag_det_res_only_clipcCs8|}d|i}t}|jjr|jjt||j}|\}}|dur&dSt j |dd}t j |dd}|}|jjrB|jj |j rVi}|||j j<|j|j|}n&|j ||jg}|jD] }|} || qf|jjr||jj i} |jdkr|d| d<|d| d<nT|jd kr|d| d <|d| d<|d | d <|d | d<n6|jdvr|d| d<n*|jdkrt|D] \} } | | d| <qn|jdkr|d| d<|d| d<nt|| |} | dd} |jjdkr|| |j} n|| |j} |jjr|jjjddt}| ||fS)Nr )Nrrr?rZf_georZf_scorerZf_borderrZf_tcor@Zf_tvo)rrrmapsrzlevel_{}rZscorerNrT)stamp)copytimer$r3r8timesstartrr,rAZ expand_dimsr[r&r0rr/runr1Z copy_from_cpuZ copy_to_cpurTr% enumerater)NotImplementedErrorr-r'rYr rXend)r9imgZori_imdatastZ shape_listZ input_dictZoutputsZ output_tensoroutputpredsiZ post_resultrUetr;r;r<__call__sl                        zTextDetector.__call__N) __name__ __module__ __qualname__r=rIrQrXrYrkr;r;r;r<r%s   r__main__T)exist_ok)r r r@rzerror in loading image:{}r_ cCg|]}|qSr;tolist.0xr;r;r< Arz cCrtr;rurwr;r;r<rzDr{z {}_{} The predict time of {}: {}z{} The predict time of {}: {}Zpngz.pdfz.pngz det_res_{}z The visualized image saved in {}zdet_results.txtw)Ur5r*pathdirnameabspath__file____dir__rTinsertjoinenvironZcv2numpyrAr]Ztools.infer.utilityZinferr.Zppocr.utils.loggingrZppocr.utils.utilityrrZ ppocr.datarrZppocr.postprocessrjsonr#objectrrl parse_argsr$Z image_dirZimage_file_listZ text_detector total_timeZdraw_img_save_dirmakedirsr"randomuniformZastypeZuint8rdrJriresZ save_resultsraidxZ image_fileZflag_gifZflag_pdfZimreaddebugr)ZimgsZpage_numlenindexrfrUrrZelapsebasenamestrdumpsZ save_predr(Zdraw_text_det_resZsrc_imZ save_filereplaceZimg_pathZimwriteopenf writelinescloser3r8reportr;r;r;r<s       y           $  C