Google I/O is a wrap! Catch up on TensorFlow sessions View sessions

tflite_support.task.vision.ImageSegmenter

Class that performs segmentation on images.

Methods

create_from_file

View source

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

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

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.