o
    'j                     @   s   d dl mZ d dl mZ d dl mZ d dlZd dlZejeZ	ej
e	 ej
eje	d d dlmZ G dd deZdS )	    )absolute_import)division)print_functionNz..)PGNet_PostProcessc                   @   s$   e Zd ZdZ	dddZdd ZdS )PGPostProcessz%
    The post process for PGNet.
    Nc                 K   sN   || _ || _|| _|| _|| _d| _tjjdkr#tjj	dkr%d| _d S d S d S )NF      T)
character_dict_path	valid_setscore_threshmodepoint_gather_modeZis_python35sysversion_infomajorminor)selfr	   r
   r   r   r   kwargs r   k/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddleocr/ppocr/postprocess/pg_postprocess.py__init__!   s   
zPGPostProcess.__init__c                 C   s>   t | j| j| j||| jd}| jdkr| }|S | }|S )N)r   fast)r   r	   r
   r   r   r   Zpg_postprocess_fastZpg_postprocess_slow)r   Z	outs_dictZ
shape_listpostdatar   r   r   __call__3   s   
zPGPostProcess.__call__)N)__name__
__module____qualname____doc__r   r   r   r   r   r   r      s
    	
r   )
__future__r   r   r   osr   pathdirname__file____dir__appendjoinZ$ppocr.utils.e2e_utils.pgnet_pp_utilsr   objectr   r   r   r   r   <module>   s   