o 0j@sddlmZddlmZmZddlmZmZmZm Z ddl m Z ddl m Z ddlmZmZmZdd lmZGd d d ee d Zd S)) annotations)ABCabstractmethod)AnyDictOptionalUnion)logging)AutoRegisterABCMetaClass) BasePredictor HPIConfigPaddlePredictorOption)uses_server_backendcsteZdZdZdZdddddddd)fddZeddZed*ddZ d+d d!Z d,d#d$Z d%d&Z d'd(Z ZS)- BasePipelinezBase class for all pipelines. This class serves as a foundation for creating various pipelines. It includes common attributes and methods that are shared among all pipeline implementations. TNF)deviceengine engine_config pp_optionuse_hpip hpi_configr Optional[str]rrOptional[Dict[str, Any]]rOptional[PaddlePredictorOption]rboolr*Optional[Union[Dict[str, Any], HPIConfig]]returnNonec s2t||_||_||_||_||_||_dS)a Initializes the class with specified parameters. Args: device (Optional[str], optional): The device to use for prediction. Defaults to `None`. engine (Optional[str], optional): Inference engine. Defaults to `None`. engine_config (Optional[Dict[str, Any]], optional): Engine-specific config. Defaults to `None`. pp_option (Optional[PaddlePredictorOption], optional): Paddle predictor options. Defaults to `None`. use_hpip (bool, optional): Whether to use HPIP. Defaults to `False`. hpi_config (Optional[Union[Dict[str, Any], HPIConfig]], optional): HPIP configuration. Defaults to `None`. N)super__init__rrrrrr)selfrrrrrrkwargs __class__a/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/pipelines/base.pyr $s  zBasePipeline.__init__cKstd)z Declaration of an abstract method. Subclasses are expected to provide a concrete implementation of `predict`. Args: input: The input data to predict. **kwargs: Additional keyword arguments. z2The method `predict` has not been implemented yet.)NotImplementedErrorr!inputr"r%r%r&predictDs zBasePipeline.predictconfigDict[str, Any]inherited_engine allow_genaituple[Optional[str], bool]cCsPd|vr|dd}||dufSd|v}|r t|ddr d}|r$dS|dfS)aResolve the effective child engine. Returns a tuple of `(engine, suppress_inherited_engine_defaults)`. Same-level `engine` has the highest priority. If a child omits `engine` but specifies another engine selector such as `use_hpip`, or a `genai_config` that targets a remote server backend, that selector should beat the inherited parent `engine` and fall back to local auto-resolution instead of reusing the parent engine defaults. rNr genai_configT)NTF)getr)r+r-r.Z child_engineZhas_local_engine_selectorr%r%r&_resolve_child_engineOs  z"BasePipeline._resolve_child_enginerr c KsHd|vr t|d|dd}|j||jdd\}}|dd}|jdur6|s6t|j}|r4|||}|d|j}|dd} |jdurj| pKi} t |jd rY|jj dd n t |jtrb|jni} i| | } d d l m } td |d||f|jdur|jnd} | d|d||j|||dd| || |ddd |S)zK Create a model instance based on the given configuration. Zmodel_config_error model_dirNTr.rrr model_dumpZ exclude_noner )create_predictorzCreating model: %s model_name batch_sizer0) r8r3rrrr9rrrr0r%) ValueErrorr1r2rrdictupdaterrhasattrr5 isinstancemodelsr7r inforcopyr) r!r+r"r3Z model_engine"suppress_inherited_engine_defaultsZmodel_engine_configZmergedrrbaser7rr%r%r& create_modeljsP             zBasePipeline.create_model'BasePipeline'c Ksd|vr t|dddlm}|j||jdd\}}|d|j}|dd}|jdurO|p0i}t|jd r>|jj d d n t |jt rG|jni}i||}|d ||j ||rYdn|j |jdure|jnd||d |S)zI Creates a pipeline based on the provided configuration. Zpipeline_config_errorr:)create_pipelineFr4rrNr5Tr6)r+rrrrrrr%)r;rGr2rr1rrr>r5r?r<rrrrB) r!r+r"rGZpipeline_enginerCrrrDr%r%r&rGs2        zBasePipeline.create_pipelinecCsdS)Nr%)r!r%r%r&closeszBasePipeline.closecKs|j|fi|S)aB Calls the `predict` method with the given input and keyword arguments. Args: input: The input data to be predicted. **kwargs: Additional keyword arguments to be passed to the `predict` method. Returns: The prediction result from the `predict` method. )r*r(r%r%r&__call__s zBasePipeline.__call__)rrrrrrrrrrrrrr)r+r,r-rr.rrr/)r+rrr )r+rrrF)__name__ __module__ __qualname____doc__Z_BasePipeline__is_baser rr* staticmethodr2rErGrIrJ __classcell__r%r%r#r&rs$    3%r) metaclassN) __future__rabcrrtypingrrrrutilsr Zutils.subclass_registerr r@r rrZmodels.common.genairrr%r%r%r&s