Class that performs segmentation on images.
tflite_support
.
task
.
vision
.
ImageSegmenter
(
options
:
tflite_support
.
task
.
vision
.
ImageSegmenterOptions
,
segmenter
:
_CppImageSegmenter
)
->
None
Methods
create_from_file
@classmethodcreate_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
@classmethodcreate_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
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.
