o yq i@s|ddlmZmZddlmZmZmZmZddlm Z ddl m Z ddl m Z ddlmZdd lmZGd d d ee d Zd S))ABCabstractmethod)AnyDictOptionalUnion)logging)AutoRegisterABCMetaClass) BasePredictor) HPIConfig)PaddlePredictorOptionc seZdZdZdZ    ddedededee e ee fe fd df fd d Z ed d Zde d efddZde fddZddZddZZS) 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. TNFdevice pp_optionuse_hpip hpi_configreturncs&t||_||_||_||_dS)ai Initializes the class with specified parameters. Args: device (str, optional): The device to use for prediction. Defaults to None. pp_option (PaddlePredictorOption, optional): The options for PaddlePredictor. Defaults to None. use_hpip (bool, optional): Whether to use the high-performance inference plugin (HPIP) by default. Defaults to False. hpi_config (Optional[Union[Dict[str, Any], HPIConfig]], optional): The default high-performance inference configuration dictionary. Defaults to None. N)super__init__rrrr)selfrrrrargskwargs __class__g/home/app/PaddleOCR-VL/.venv_paddleocr/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.)NotImplementedErrorrinputrrrrpredict>s zBasePipeline.predictconfigc Ksd|vr t|d|dd}|d|j}|dd}|jdur-|p%i}i|j|}|dd}ddlm}td |d |f|jdurN|j }nd}|d|d ||j |d d ||||d |} | S)a. Create a model instance based on the given configuration. Args: config (Dict): A dictionary containing configuration settings. **kwargs: The model arguments that needed to be pass. Returns: BasePredictor: An instance of the model. Zmodel_config_error model_dirNrr genai_configr )create_predictorzCreating model: %s model_name batch_size)r&r#rr'rrrr$r) ValueErrorgetrrr%r inforcopyr) rr"rr#rrr$r%rmodelrrr create_modelIs6           zBasePipeline.create_modelcCsd|vr t|dddlm}|d|j}|dd}|jdur-|p%i}i|j|}|||j|jdur;|jn|j||d}|S)z Creates a pipeline based on the provided configuration. Args: config (Dict): A dictionary containing the pipeline configuration. Returns: BasePipeline: An instance of the created pipeline. Zpipeline_config_errorr()create_pipelinerrN)r"rrrr) r)r+r0r*rrrrr-)rr"r0rrZpipelinerrrr0ws      zBasePipeline.create_pipelinecCsdS)Nr)rrrrcloseszBasePipeline.closecKs|j|fi|S)a< 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!rrrr__call__s zBasePipeline.__call__)NNFN)__name__ __module__ __qualname____doc__Z_BasePipeline__is_basestrrboolrrrrr rrr!r r/r0r1r2 __classcell__rrrrrs0  . r) metaclassN)abcrrtypingrrrrutilsr Zutils.subclass_registerr modelsr Z utils.hpir Zutils.pp_optionrrrrrrs