o 1j,@sddlmZddlmZddlmZddlmZddlm Z m Z m Z ddl m Z ddlmZd d lmZd ed d fddZGdd d eedZdS))ABC)Path)logging)AttrDict)check_supported_deviceset_env_for_deviceupdate_device_num)FLAGS_json_format_model)AutoRegisterABCMetaClass) build_modelconfigreturn BaseExportorcCs.|jj}zWn tyYnwt||S)zbuild model exportor Args: config (AttrDict): PaddleX pipeline config, which is loaded from pipeline yaml file. Returns: BaseExportor: the exportor, which is subclass of BaseExportor. )GlobalmodelModuleNotFoundErrorrget)rZ model_namer^/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/modules/base/exportor.pybuild_exportors  rcs^eZdZdZdZfddZddZdefdd Zdd e de fd d Z ddZ ddZ ZS)rzBase Model ExportorTcs`t|j|_|j|_||jj}|jddr"|jdd}t |jj |d\|_ |_ dS)zInitialize the instance. Args: config (AttrDict): PaddleX pipeline config, which is loaded from pipeline yaml file. Zbasic_config_pathN) config_path) super__init__r global_configZExport export_configget_config_path weight_pathrr rZ pdx_config pdx_model)selfrr __class__rrr4s zBaseExportor.__init__cCsVt|jd}z|}Wn tyd}Ynw|s)td|d|dd}|S)z get config path Args: weight_path (str): The path to the weight Returns: config_path (str): The path to the config z config.yamlFzThe config file(`z`) related to weight file(`z%`) is not exist, use default instead.N)rparentexists Exceptionrwarning)r rrZ is_existsrrrrFs   zBaseExportor.get_config_pathrcCs>||jjdi|}|jdksJd|jddS)zWexecute model exporting Returns: dict: the export metrics rz Encountered an unexpected error(z) in exporting!Nr) update_configrexportget_export_kwargs returncode)r Z export_resultrrrr(`s   zBaseExportor.exportNusing_device_numbercCsHt|jj|jjt|jj|rt|jj|n|jj}|dd}|S)a#get device setting from config Args: using_device_number (int, optional): specify device number to use. Defaults to None, means that base on config setting. Returns: str: device setting, such as: `gpu:0,1`, `npu:0,1`, `cpu`. ZdcuZgpu)rrdevicerrr replace)r r+Zdevice_settingrrr get_deviceos  zBaseExportor.get_devicecCsdS)zupdate export configNr)r rrrr'szBaseExportor.update_configcCs.|jddpt}|jj|jj|d|dS)z0get key-value arguments of model export functionexport_with_pirFr )rsave_dirr,r/)rrr rroutputr.)r r/rrrr)szBaseExportor.get_export_kwargs)N)__name__ __module__ __qualname____doc__Z_BaseExportor__is_baserrdictr(intstrr.r'r) __classcell__rrr!rr/s ) metaclassN)abcrpathlibrutilsrZ utils.configrZ utils.devicerrr Z utils.flagsr Z utils.miscr r rrrrrrs