o 0j@sddlmZmZmZmZmZddlZddlm Z ddl m Z m Z ddl mZddlmZd d lmZeje d Gd d d eZdS))AnyDictListOptionalUnionN)pipeline_requires_extra) HPIConfigPaddlePredictorOption) DetResult) benchmark) BasePipelineZ multimodalcseZdZdZdZddddddddedeedeed eeeefd ee d e d ee eeefe fd dffddZ dde eeeejeejfdede eeefdfd efddZZS)OpenVocabularyDetectionPipelinez"Open Vocabulary Detection PipelineZopen_vocabulary_detectionNFdeviceengine engine_config pp_optionuse_hpip hpi_configconfigrrrrrrreturnc sRtjd||||||d||didddi} || |_| d|_dS) a#Initializes the open vocabulary detection pipeline. Args: config (Dict): Configuration dictionary containing model and other parameters. 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`. rZ SubModulesZOpenVocabularyDetectionZmodel_config_errorz(config error for doc_ori_classify_model! thresholdsN)super__init__getZ create_modelopen_vocabulary_detection_modelr) selfrrrrrrrkwargsZ&open_vocabulary_detection_model_config __class__r/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/pipelines/open_vocabulary_detection/pipeline.pyr!s$ z(OpenVocabularyDetectionPipeline.__init__inputpromptrcks|j|||dEdHdS)a}Predicts open vocabulary detection results for the given input. Args: input (Union[str, list[str], np.ndarray, list[np.ndarray]]): The input image(s) or path(s) to the images. prompt (str): The text prompt used to describe the objects. thresholds (dict | None): Threshold values for different models. If provided, these will override any default threshold values set during initialization. Default is None. **kwargs: Additional keyword arguments that can be passed to the function. Returns: DetResult: The predicted open vocabulary detection results. )r&rN)r)r r%r&rr!rrr$predictMsz'OpenVocabularyDetectionPipeline.predict)N)__name__ __module__ __qualname____doc__entitiesrrstrrr boolrr rrnpZndarrayfloatr r' __classcell__rrr"r$rsH  0r)typingrrrrrnumpyr/Z utils.depsrmodelsr r Zmodels.object_detection.resultr Zutils.benchmarkr baserZ time_methodsrrrrr$s