=pjSSKJrJrJrJrJr SSKrSSKJ r SSK J r J r SSK Jr SSKJr S S KJr S S KJr \R*"S S \55r\ "S5"SS\55rg))AnyDictListOptionalUnionN)pipeline_requires_extra) HPIConfigPaddlePredictorOption)InstanceSegResult) benchmark)(AutoParallelImageSimpleInferencePipeline) BasePipelinec^\rSrSrSrSSSSSSS.S\S\\S\\S \\\\4S \\ S \ S \\ \\\4\ 4S S4U4Sjjjr SS\ \\\\R \\R 4S\ \S4S \4SjjrSrU=r$)_InstanceSegmentationPipelinezInstance Segmentation PipelineNFdeviceengine engine_config pp_optionuse_hpip hpi_configconfigrrrrrrreturnc >[T U]"SUUUUUUS.UD6 USSn URU 5UlU SUlg)aInitializes the instance 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`. r SubModulesInstanceSegmentation thresholdN)super__init__ create_modelinstance_segmentation_modelr!) selfrrrrrrrkwargs"instance_segmentation_model_config __class__s v/var/www/html/pdf-tiff/venv/lib/python3.13/site-packages/paddlex/inference/pipelines/instance_segmentation/pipeline.pyr$&_InstanceSegmentationPipeline.__init__sn2  '!   .4L-A ". *,0+<+< ., (# URXS9ShvN gN7f)aPredicts instance 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. threshold (Union[float, None]): The threshold value to filter out low-confidence predictions. Default is None. **kwargs: Additional keyword arguments that can be passed to the function. Returns: InstanceSegResult: The predicted instance segmentation results. )r!N)r&)r'r.r!r(s r+predict%_InstanceSegmentationPipeline.predictJs 33E3OOOs )r&r!N)__name__ __module__ __qualname____firstlineno____doc__rrstrrr boolrr r$rnpndarrayfloatr r0__static_attributes__ __classcell__)r*s@r+rrs( !% $2659AE)I)I )I  )I  S#X/ )I12)I)IU4S>9#<=>)I )I)I\)-PS$s)RZZbjj1AABP%P  PPr-rcvc.\rSrSrSr\S5rSrSrg)InstanceSegmentationPipeline]instance_segmentationc[$r2)r)r's r+ _pipeline_cls*InstanceSegmentationPipeline._pipeline_clsas,,r-c2USSRSS5$)Nrr batch_size)get)r'rs r+_get_batch_size,InstanceSegmentationPipeline._get_batch_sizees l#$:;?? aPPr-r"N) r3r4r5r6entitiespropertyrErKr=r"r-r+rArA]s&H --Qr-rA)typingrrrrrnumpyr: utils.depsr modelsr r #models.instance_segmentation.resultr utils.benchmarkr _parallelrbaser time_methodsrrAr"r-r+rXsk4326D(@ >PL>P>PBQ#KQQr-