o e@s0ddlmZmZddlmZGdddeZdS))SEMANTIC_SEGMENTATION_MODELSEMANTIC_SEGMENTATION_URL)InferenceModelcsJeZdZdZdedeffdd Zd dedeffd d Zd d ZZ S)SemanticSegmentationModelzq Run inference on a semantic segmentation model hosted on Roboflow or served through Roboflow Inference. api_key version_idcs0tt|||td|jd|j|_dS)z Create a SemanticSegmentationModel object through which you can run inference. Args: api_key (str): private roboflow api key version_id (str): the workspace/project id /N)superr__init__r dataset_idversionapi_url)selfrr __class__T/var/www/myenv/lib/python3.10/site-packages/roboflow/models/semantic_segmentation.pyr sz"SemanticSegmentationModel.__init__2 image_path confidencecstt|j||tdS)ac Infers detections based on image from a specified model and image path. Args: image_path (str): path to the image you'd like to perform prediction on confidence (int): confidence threshold for predictions, on a scale from 0-100 Returns: PredictionGroup Object Example: >>> import roboflow >>> rf = roboflow.Roboflow(api_key="") >>> project = rf.workspace().project("PROJECT_ID") >>> model = project.version("1").model >>> prediction = model.predict("YOUR_IMAGE.jpg") )rprediction_type)r rpredictr)rrrrrrrs z!SemanticSegmentationModel.predictcCs"dt|jd|jd|jdS)N)type__name__idr )rrrr__str__1s"z!SemanticSegmentationModel.__str__)r) r __module__ __qualname____doc__strr intrr __classcell__rrrrrs  rN)roboflow.configrrroboflow.models.inferencerrrrrrs