
    i              	       p   S SK r S SKr\ R                  R                  \ R                  R	                  \5      5      r\R                  R                  \5        \R                  R                  S \ R                  R	                  \ R                  R                  \S5      5      5        S\ R                  S'   S SKrS SKrS SKrS SKrS SKJs  Jr  S SKJr  S SKJrJr  S SKJrJr  S SKJr  \" 5       r " S	 S
\5      r\S:X  Ga  \R@                  " 5       r!\" \!RD                  5      r#\" \!5      r$S r%S r&Sr'\ R                  RQ                  \'5      (       d  \ RR                  " \'5        \# GH  r*\" \*5      u  r+r,r-\,(       d  \R\                  " \*5      r+\+c"  \R_                  SRa                  \*5      5        MN  \$" \+5      u  r1r2r3\%S :  a  \&\3-  r&\%S-  r%\R_                  SRa                  \*\35      5        \Rh                  " \1\2\*5      r5\ R                  Rm                  \*5      S   r7\ R                  R                  \'SRa                  \75      5      r8\Rr                  " \8\55        \R_                  SRa                  \85      5        GM     \%S:  a'  \R_                  SRa                  \&\%S-
  -  5      5        ggg)    Nz../..auto_growthFLAGS_allocator_strategy)
get_logger)get_image_file_listcheck_and_read)create_operators	transform)build_post_processc                   ,    \ rS rSrS rS rS rS rSrg)TextE2E%   c                    Xl         UR                  U l        UR                  U l        S0 0S/ SQ/ SQSSS.0SS 0S	S
SS/00/n0 nU R                  S:X  aV  SUR                  SS.0US'   SUS'   UR                  US'   UR
                  US'   UR                  US'   UR                  US'   OD[        R                  SR                  U R                  5      5        [        R                  " S5        [        U5      U l        [        U5      U l        ["        R$                  " US[        5      u  U l        U l        U l        ng )NE2EResizeForTestNormalizeImage)gZd;O?gy&1?g?)g
ףp=
?gv/?gCl?z1./255.hwc)stdmeanscaleorder
ToCHWImageKeepKeys	keep_keysimageshapePGNet	totaltext)max_side_len	valid_setr   PGPostProcessnamescore_threshcharacter_dict_pathr   modezunknown e2e_algorithm:{}e2e)argse2e_algorithmuse_onnxe2e_limit_side_lene2e_pgnet_score_threshe2e_char_dict_pathe2e_pgnet_valid_sete2e_pgnet_modeloggerinfoformatsysexitr   preprocess_opr
   postprocess_oputilitycreate_predictor	predictorinput_tensoroutput_tensors)selfr%   pre_process_listpostprocess_params_s        a/var/www/html/banglarbhumi/venv/lib/python3.13/site-packages/paddleocr/tools/infer/predict_e2e.py__init__TextE2E.__init__&   sc   	!//$ 01&"	# 4 +'9:;
  ("$($;$;!,%#Q *9v&151L1L~.8<8O8O45.2.F.F{+)-)<)<v&KK299$:L:LMNHHQK-.>?01CD $$%
	
N    c           
          [        UR                  S   5       HW  n[        [        [	        XS4   S5      US-
  5      5      XS4'   [        [        [	        XS4   S5      US-
  5      5      XS4'   MY     U$ )Nr      )ranger   intminmax)r9   points
img_height	img_widthpnos        r=   clip_det_resTextE2E.clip_det_resT   sp    a)C SQ%;Y]!KLF6N SQ%;Z!^!LMF6N * r@   c                     USS u  p4/ nU H&  nU R                  XcU5      nUR                  U5        M(     [        R                  " U5      nU$ )Nr      )rK   appendnparray)r9   dt_boxesimage_shaperH   rI   dt_boxes_newboxs          r=   filter_tag_det_res_only_clip$TextE2E.filter_tag_det_res_only_clipZ   sU     +Aa 0
C##CY?C$  88L)r@   c                    UR                  5       nSU0n[        X0R                  5      nUu  pUc  g[        R                  " USS9n[        R                  " USS9nUR                  5       n[
        R
                  " 5       nU R                  (       ac  0 nXU R                  R                  '   U R                  R                  U R                  U5      n0 nUS   US'   US   US'   US   US	'   US
   US'   OU R                  R                  U5        U R                  R                  5         / nU R                   H$  n	U	R                  5       n
UR                  U
5        M&     0 nU R                  S:X  a!  US   US'   US   US'   US   US	'   US
   US'   O[         eU R#                  X5      nUS   US   pU R%                  XR&                  5      n[
        R
                  " 5       U-
  nXU4$ )Nr   )Nr   r   )axisf_borderrB   f_charrN   f_direction   f_scorer   rG   texts)copyr	   r2   rP   expand_dimstimer'   r7   r    r6   runr8   copy_from_cpucopy_to_cpurO   r&   NotImplementedErrorr3   rV   r   )r9   imgori_imdata
shape_list	starttime
input_dictoutputspredsoutput_tensoroutputpost_resultrG   strsrR   elapses                   r=   __call__TextE2E.__call__c   s   ~112;nnSq)^^JQ7
hhjIIK	==J14t((--.nn(()<)<jIGE '
E*%ajE(O#*1:E- &qzE)++C0NN G!%!4!4&224v& "5 E!!W,$+AJj!")!*h'.qzm$#*1:i ))))%<"8,k'.B44V\\Jy(v%%r@   )r%   r&   r7   r8   r3   r6   r2   r'   N)	__name__
__module____qualname____firstlineno__r>   rK   rV   rt   __static_attributes__ r@   r=   r   r   %   s    +
\)&r@   r   __main__z./inference_resultszerror in loading image:{}rB   zPredict time of {}: {}z
e2e_res_{}z The visualized image saved in {}zAvg Time: {}):osr0   pathdirnameabspath__file____dir__rO   insertjoinenvironcv2numpyrP   rb   tools.infer.utilityinferr4   ppocr.utils.loggingr   ppocr.utils.utilityr   r   
ppocr.datar   r	   ppocr.postprocessr
   r-   objectr   rv   
parse_argsr%   	image_dirimage_file_listtext_detectorcount
total_timedraw_img_saveexistsmakedirs
image_filerg   flagr<   imreadr.   r/   rG   rr   rs   draw_e2e_ressrc_imsplitimg_name_pureimg_pathimwriter{   r@   r=   <module>r      s=   
 

''//"''//(3
4   277??277<<#AB C)6

% & 
   
 % % * C 2 0	g&f g&T zD)$..9ODMMEJ)M77>>-((
M"%
%j1T1**Z(C;KK3:::FG,S1f19& J
,33JGH%%fdJ?j1"577<<|/B/B=/QRHf%6==hGH! &" qyN))*	*BCD 5 r@   