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
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