o
    e                     @   s   d dl Z d dlZd dlZdd Zdd Zedkrfe Zejde	dd	d
 ejde	ddd
 ejde	ddd
 ejde	ddd
 e
 ZejdkrVed eejej dS ejdkrheejejej dS dS dS )    Nc              	   C   s   t |dS}t | d5}| D ](}|dddd}|d |d }}|d	d}||d
 | d  qW d    n1 sCw   Y  W d    d S W d    d S 1 s[w   Y  d S )Nwr
  ,r      "	)open	readlinesstripreplacesplitwrite)
input_path	out_labelout_fileflinetmpimg_pathlabel r   OD:\Projects\ConvertPro\env\Lib\site-packages\paddleocr/ppocr/utils/gen_label.pygen_rec_label   s   "r   c              
   C   s<  t |d}t|D ]}| |dd  d }g }t tj||dddN}| D ]A}|dd	d
d}	|	d d }
g }t	dt
|
dD ]}|
||d  }dd |D }|| qH|	d |d}|| q*W d    n1 svw   Y  ||d tj|dd d  qW d    d S 1 sw   Y  d S )Nr      z.jpgr   z	utf-8-sig)encodingz
u   ï»¿r   r      r      c                 S   s   g | ]}t |qS r   )int).0tr   r   r   
<listcomp>,   s    z!gen_det_label.<locals>.<listcomp>)Ztranscriptionpointsr
   F)ensure_asciir   )r   oslistdirpathjoinr   r   r   r   rangelenappendr   jsondumps)	root_pathZ	input_dirr   r   Z
label_filer   r   r   r   r   r%   sibresultr   r   r   gen_det_label   s@   "r5   __main__z--modeZrecz6Generate rec_label or det_label, can be set rec or det)typedefaulthelpz--root_path.z=The root directory of images.Only takes effect when mode=det z--input_pathz)Input_label or input path to be convertedz--output_labelzout_label.txtzOutput file namezGenerate rec labelZdet)r'   argparser.   r   r5   __name__ArgumentParserparseradd_argumentstr
parse_argsargsmodeprintr   Zoutput_labelr0   r   r   r   r   <module>   sN   


