o
    H!i	                     @   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   Z/home/app/PaddleOCR-VL-test/.venv_paddleocr/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rS| j D ]}trBq=|	  | j
rP|j| j
d q=dS t| j  | j dS )zthe main functioncheck_datasettrainevaluateexportpredict)Z	save_pathN)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   