
    i                     "    S SK r  " S S\5      rg)    Nc                   :   ^  \ rS rSrSrSU 4S jjrSS jrSrU =r$ )ClsPostProcess   z)Convert between text-label and text-indexc                 B   > [         [        U ]  5         Xl        X l        g N)superr   __init__
label_listkey)selfr
   r   kwargs	__class__s       k/var/www/html/banglarbhumi/venv/lib/python3.13/site-packages/paddleocr/ppocr/postprocess/cls_postprocess.pyr	   ClsPostProcess.__init__   s    nd,.$    c                    U R                   b  XR                      nU R                  nUc(  [        UR                  S   5       Vs0 s H  ofU_M     nn[	        U[
        R                  5      (       a  UR                  5       nUR                  SS9n[        U5       VVs/ s H  u  pXV   XU4   4PM     n	nnUc  U	$ U Vs/ s H	  oeU   S4PM     nnX4$ s  snf s  snnf s  snf )N   )axisg      ?)
r   r
   rangeshape
isinstancepaddleTensornumpyargmax	enumerate)
r   predslabelargsr   r
   idx	pred_idxsi
decode_outs
             r   __call__ClsPostProcess.__call__   s    88((OE__
.3EKKO.DE.Dss(.DJEeV]]++KKMELLaL(	;DY;O
;OZ_esFm,;O 	 
 =3895CS/3'59   F

 :s   CCC!)r   r
   )NNr   )	__name__
__module____qualname____firstlineno____doc__r	   r%   __static_attributes____classcell__)r   s   @r   r   r      s    3
! !r   r   )r   objectr    r   r   <module>r0      s    !V !r   