Options for the image segmenter task.
mp.tasks.vision.ImageSegmenterOptions(
base_options: mp.tasks.BaseOptions
,
running_mode: mp.tasks.vision.RunningMode
= mp.tasks.vision.FaceDetectorOptions.running_mode
,
output_confidence_masks: bool = True,
output_category_mask: bool = False,
result_callback: Optional[Callable[[ImageSegmenterResult, image_module.Image, int], None]] = None
)
Attributes |
base_options
|
Base options for the image segmenter task.
|
running_mode
|
The running mode of the task. Default to the image mode. Image
segmenter task has three running modes: 1) The image mode for segmenting
objects on single image inputs. 2) The video mode for segmenting objects
on the decoded frames of a video. 3) The live stream mode for segmenting
objects on a live stream of input data, such as from camera.
|
output_confidence_masks
|
Whether to output confidence masks.
|
output_category_mask
|
Whether to output category mask.
|
result_callback
|
The user-defined result callback for processing live stream
data. The result callback should only be specified when the running mode
is set to the live stream mode.
|
Methods
__eq__
__eq__(
other
)
Class Variables |
output_category_mask
|
False
|
output_confidence_masks
|
True
|
result_callback
|
None
|
running_mode
|
<VisionTaskRunningMode.IMAGE: 'IMAGE'>
|