tflite_support.task.processor.SegmentationOptions

Options for segmentation processor.

tflite_support.task.processor.SegmentationOptions(
    display_names_locale: Optional[str] = None,
    output_type: Optional[tflite_support.task.processor.OutputType] = tflite_support.task.processor.OutputType.CATEGORY_MASK
)

display_names_locale The locale to use for display names specified through the TFLite Model Metadata.
output_type The output mask type allows specifying the type of post-processing to perform on the raw model results.

Methods

create_from_pb2

View source

@classmethod
create_from_pb2(
    pb2_obj: _SegmentationOptionsProto
) -> 'SegmentationOptions'

Creates a SegmentationOptions object from the given protobuf object.

to_pb2

View source

to_pb2() -> _SegmentationOptionsProto

Generates a protobuf object to pass to the C++ layer.

__eq__

View source

__eq__(
    other: Any
) -> bool

Checks if this object is equal to the given object.

Args
other The object to be compared with.

Returns
True if the objects are equal.

display_names_locale None
output_type <OutputType.CATEGORY_MASK: 1>