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

create_from_file

View source

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

create_from_options

View source

@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

View source

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.