tflite_support.task.processor.ColoredLabel

Defines a label associated with an RGB color, for display purposes.

tflite_support.task.processor.ColoredLabel(
    color: Tuple[int, int, int], category_name: str, display_name: str
)

color The RGB color components for the label, in the [0, 255] range.
category_name The class name, as provided in the label map packed in the TFLite ModelMetadata.
display_name The display name, as provided in the label map (if available) packed in the TFLite Model Metadata .

Methods

create_from_pb2

View source

@classmethod
create_from_pb2(
    pb2_obj: _ColoredLabelProto
) -> 'ColoredLabel'

Creates a ColoredLabel object from the given protobuf object.

to_pb2

View source

to_pb2() -> _ColoredLabelProto

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.