o 0ju@sddlmZmZmZmZmZddlZddlm Z ddl m Z m Z ddl mZddlmZd d lmZeeefZeje d Gd d d eZdS))AnyDictListOptionalUnionN)pipeline_requires_extra) HPIConfigPaddlePredictorOption) SAMSegResult) benchmark) BasePipelineZ multimodalcseZdZdZdZddddddddedeedeed eeeefd ee d e d ee eeefe fd dffddZ dde eeeejeejfde eeeejfded efddZZS)"OpenVocabularySegmentationPipelinez%Open Vocabulary Segmentation pipelineZopen_vocabulary_segmentationNFdeviceengine engine_config pp_optionuse_hpip hpi_configconfigrrrrrrreturnc sltjd||||||d||didddi} || |_|didddi} || |_dS) a&Initializes the open vocabulary segmentation 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 SubModulesZBoxPromptSegmentationZmodel_config_errorz(config error for doc_ori_classify_model!ZPointPromptSegmentationN)super__init__getZ create_modelbox_prompted_modelpoint_prompted_model) selfrrrrrrrkwargsZbox_prompted_model_cfgZpoint_prompted_model_cfg __class__r/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/pipelines/open_vocabulary_segmentation/pipeline.pyr#s(   z+OpenVocabularySegmentationPipeline.__init__boxinputprompt prompt_typecksR|dkr|j|d|idEdHdS|dkr%|j|d|idEdHdStd)a#Predicts image segmentation results for the given input. Args: input (str | list[str] | np.ndarray | list[np.ndarray]): The input image(s) or path(s) to the images. prompt (list[list[float]] | np.ndarray): The prompt for the input image(s). prompt_type (str): The type of prompt, either 'box' or 'point'. Default is 'box'. **kwargs: Additional keyword arguments that can be passed to the function. Returns: SAMSegResult: The predicted SAM segmentation results. r%Z box_prompt)ZpromptsNpointZ point_promptz9Invalid prompt type. Only 'box' and 'point' are supported)rr ValueError)r r&r'r(r!rrr$predictTsz*OpenVocabularySegmentationPipeline.predict)r%)__name__ __module__ __qualname____doc__entitiesrrstrrr boolrr rrnpZndarrayfloatr r+ __classcell__rrr"r$rsH  5r)typingrrrrrnumpyr3Z utils.depsrmodelsr r Z+models.open_vocabulary_segmentation.resultsr Zutils.benchmarkr baserintr4NumberZ time_methodsrrrrr$s