
    i              	       z   S SK Jr  S SK Jr  S SK Jr  S SKr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JrJr  S S	KJr  S S
KJr  S SKJr  S SKJr  S SKJ r   S SK!J"r"J#r#J$r$  S SK%r% SS jr&S r'S r(\)S:X  a  \ RT                  " 5       u  r+r,r-r.\(" 5         gg)    )absolute_import)division)print_functionNz..auto_growthFLAGS_allocator_strategy)create_operators	transform)build_model)build_post_process)
load_model)get_image_file_list)Image	ImageDraw	ImageFontc           	         U R                   U R                  pvU R                  5       n[        R                  " SXv4S5      n	SS Kn
U
R                  S5        [        R                  " U5      n[        R                  " U	5      n[        [        X5      5       H  u  nu  p[        R                  " U5      nU Vs/ s H  n[        U5      PM     nnU
R                  SS5      U
R                  SS5      U
R                  SS5      4nUR                  UUS9  UR                  UUS9  [         R"                  " USSS	9nUR%                  US   S   US   S
   /USUS9  M     [        R&                  " XS5      n[        R                  " SUS-  U4S5      nUR)                  USSXv45        UR)                  XSUS-  U45        [*        R,                  R/                  US   S   5      S-   n[*        R,                  R1                  U5      (       d  [*        R2                  " U5        [*        R,                  R5                  U[*        R,                  R7                  U5      5      n[8        R:                  " U[        R                  " U5      S S 2S S 2S S S24   5        [<        R?                  SRA                  U5      5        g s  snf )NRGB)   r   r   r   r   )fill)outline   zutf-8)encoding   r   r   r   )r   fontg      ?   Globalsave_res_path/e2e_results/The e2e Image saved in {})!heightwidthcopyr   newrandomseedr   Draw	enumeratezipnparraytuplerandintpolygonr   truetypetextblendpasteospathdirnameexistsmakedirsjoinbasenamecv2imwriteloggerinfoformat)imageboxestxtsconfigimg_name	font_pathhwimg_left	img_rightr%   	draw_left
draw_rightidxboxtxtxcolorr   img_showsave_e2e_path	save_paths                         Y/var/www/html/banglarbhumi/venv/lib/python3.13/site-packages/paddleocr/tools/infer_e2e.pydraw_e2e_res_for_chineserT   ,   s,    <<qzz|H		%!9I
KKNx(I	*J$S%56Zchhsm!$%AuQx%3'3)?PQSVAWX#E*3.!!)R'BQCF1I.)$O 7 {{5C0HyyQ
O<HNN8aA\*NN9!QUA./GGOOF8$4_$EFXM77>>-((
M"]BGG,<,<X,FGIKK	288H-aDbDj9:
KK+229=>! &s   8J;c                    [        U 5      S:  Gax  Un[        X5       H  u  pgUR                  [        R                  5      R                  S5      n[        R                  " XV/SSSS9  [        R                  " UU[        US   5      [        US   5      4[        R                  S	S
SS9  M     [        R                  R                  US   S   5      S-   n[        R                  R                  U5      (       d  [        R                  " U5        [        R                  R!                  U[        R                  R#                  U5      5      n	[        R$                  " X5        [&        R)                  SR+                  U	5      5        g g )Nr   )r   r   r   T)r   r   r   r   )rO   	thicknessr   )r   r   r   gffffff?)r   r   r   r   )orgfontFace	fontScalerO   rV   r   r   r   r    )lenr)   astyper*   int32reshaper:   	polylinesputTextintFONT_HERSHEY_COMPLEXr3   r4   r5   r6   r7   r8   r9   r;   r<   r=   r>   )
dt_boxesstrsrB   imgrC   src_imrL   strsave_det_pathrR   s
             rS   draw_e2e_resrh   M   s&   
8}qH+HC**RXX&..z:CMM&%]aPKKW&CL(9:11! , GGOOF8,_=>P 	 ww~~m,,KK&GGLL0@0@0JK	I&/66yAB+     c                     [         S   n [        [         S   5      n[        [         U5        [        [         S   U 5      n/ n[         S   S   S    H9  n[	        U5      S   nSU;   a  M  US	:X  a	  S
S/XE   S'   UR                  U5        M;     [        X05      n[         S   S   n[        R                  R                  [        R                  R                  U5      5      (       d3  [        R                  " [        R                  R                  U5      5        UR                  5         [        US5       n[        [         S   S   5       GH  n	[        R!                  SR#                  U	5      5        [        U	S5       n
U
R%                  5       nS
U0nS S S 5        ['        WU5      n[(        R*                  " US   SS9n[(        R*                  " US   SS9n[,        R.                  " U5      nU" U5      nU" UU5      nUS   US   nn/ n[1        UU5       H.  u  nnSU0nUR3                  5       US'   UR                  U5        M0     U	S-   [4        R6                  " U5      -   S-   nUR9                  UR;                  5       5        [<        R>                  " U	5      nU S   S:X  a  [A        UU[         UU	5        GMq  U S   S:X  d  GM}  [B        RD                  " [<        RF                  " U[<        RH                  5      5      n[K        UUU[         U	SS9  GM     S S S 5        [        R!                  S5        g ! , (       d  f       GN= f! , (       d  f       N6= f)Nr   ArchitecturePostProcessEvaldataset
transformsr   LabelKeepKeysr?   shape	keep_keysr   wb	infer_imgzinfer_img: {}rb)axisr   pointstextstranscription	
infer_visual_typeENCNz./doc/fonts/simfang.ttf)rD   zsuccess!)&rB   r
   r   r   listappendr   r3   r4   r6   r5   r7   evalopenr   r<   r=   r>   readr	   r*   expand_dimspaddle	to_tensorr)   tolistjsondumpswriteencoder:   imreadrh   r   	fromarraycvtColorCOLOR_BGR2RGBrT   )global_configmodelpost_process_classro   opop_nameopsr   foutfilefrd   databatchimages
shape_listpredspost_resultrx   rc   dt_boxes_jsonpolyrf   tmp_jsonotstrsrc_imgs                             rS   mainr   f   s   8$M ~./Evu ,F=,A=Q JVnY'5r(1+g
"(/'9BK$" 6 :
5C8$_5M77>>"''//-899
BGGOOM23	JJL	mT	"d'x(8(EFDKK..t45dD!Qffh~ " dC(E^^E!H15Faq9J%%f-F&ME,UJ?K&x0+g2FDFM .	c+S1%)[[]"$$X. / 4K$**]";;dBEJJu||~&jj&G01T9VT67DA23t;//#,,w@Q@Q*RS(73 G 
#F KK
A "! 
#	"s-   4A
M>L5D/MAM5
M?M
M__main__)z./doc/simfang.ttf)/
__future__r   r   r   numpyr*   r3   sysr4   r5   abspath__file____dir__r   insertr8   environr:   r   r   
ppocr.datar   r	   ppocr.modeling.architecturesr
   ppocr.postprocessr   ppocr.utils.save_loadr   ppocr.utils.utilityr   tools.programprogramPILr   r   r   mathrT   rh   r   __name__
preprocessrB   devicer<   
vdl_writer ri   rS   <module>r      s    '  %  	 

''//"''//(3
4   277??277<<#>? @)6

% & 
   2 4 0 , 3  + +  5H?BC2?D z)0););)=&FFFJF ri   