cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Updated docstrings for get_labels

Browse Source
This commit is contained in:
kinaryml 2023-04-18 02:54:03 -07:00
parent 723cb2a919
commit 1919b0e341
1 changed files with 0 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
mediapipe/tasks/python/vision/image_segmenter.py
View File

@ -258,17 +258,6 @@ class ImageSegmenter(base_vision_task_api.BaseVisionTaskApi):
If there is no label map provided in the model file, empty label list is If there is no label map provided in the model file, empty label list is
returned. returned.
Returns:
If the output_type is CATEGORY_MASK, the returned vector of images is
per-category segmented image mask.
If the output_type is CONFIDENCE_MASK, the returned vector of images
contains only one confidence image mask. A segmentation result object that
contains a list of segmentation masks as images.
Raises:
ValueError: If any of the input arguments is invalid.
RuntimeError: If image segmentation failed to run.
""" """
return self.labels return self.labels