o
    0jr                     @   s`   d dl mZ d dlmZmZ d dlmZ d dlmZ d dl	m
Z
 d dlmZ G dd deZd	S )
   )_ModelBasedConfig)
get_config
parse_args)raise_unsupported_api_error)INFER_BENCHMARK)disable_pir_bydefault)try_except_decoratorc                   @   s$   e Zd Zd Zdd Zedd ZdS )Enginec                 C   s:   t  }t|j|jdd}|jj| _|jj| _t	|| _
d S )NF)Z	overridesshow)r   r   configoverrideGlobalmode_modeoutput_outputr   _model)selfargsr    r   O/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/engine.py__init__   s
   

zEngine.__init__c                 C   s   | j dkr
| j S | j dkrt  | j  dS | j dkr&t  | j S | j dkr3t  | j S | j dkr=| j S | j dkr]| j D ]}t	rLqG|
  | jrZ|j| jd qGdS t| j  | j dS )	zthe main functioncheck_datasettrainevaluateexportpdparams2safetensorspredict)Z	save_pathN)r   r   r   r   r   r   r   r   r   r   printr   Zsave_allr   	__class__)r   resr   r   r   run"   s.   









z
Engine.runN)__name__
__module____qualname____doc__r   r   r!   r   r   r   r   r	      s
    r	   N)modelr   Zutils.configr   r   Zutils.errorsr   Zutils.flagsr   Zutils.lazy_loaderr   Zutils.result_saverr   objectr	   r   r   r   r   <module>   s   