Introducing LiteRT: Google's high-performance runtime for on-device AI, formerly known as TensorFlow Lite. Learn more

tflite_support.task.vision.ImageSegmenter

Class that performs segmentation on images.

tflite_support.task.vision.ImageSegmenter(
    options: tflite_support.task.vision.ImageSegmenterOptions,
    segmenter: _CppImageSegmenter
) -> None

Methods

@classmethod
create_from_file(
    file_path: str
) -> 'ImageSegmenter'

Creates the ImageSegmenter object from a TensorFlow Lite model.

Args
`file_path`	Path to the model.

Returns
`ImageSegmenter` object that's created from `options`.

Raises
`ValueError`	If failed to create `ImageSegmenter` object from the provided file such as invalid file.
`RuntimeError`	If other types of error occurred.

@classmethod
create_from_options(
    options: tflite_support.task.vision.ImageSegmenterOptions
) -> 'ImageSegmenter'

Creates the ImageSegmenter object from image segmenter options.

Args
`options`	Options for the image segmenter task.

Returns
`ImageSegmenter` object that's created from `options`.

Raises
`ValueError`	If failed to create `ImageSegmenter` object from `ImageSegmenterOptions` such as missing the model.
`RuntimeError`	If other types of error occurred.

segment(
    image: tflite_support.task.vision.TensorImage
) -> tflite_support.task.processor.SegmentationResult

Performs segmentation on the provided TensorImage.

Args
`image`	Tensor image, used to extract the feature vectors.

Returns
segmentation result.

Raises
`ValueError`	If any of the input arguments is invalid.
`RuntimeError`	If failed to run segmentation.