tflite_support.task.processor.Category

A classification category.

tflite_support.task.processor.Category(
    index: int, score: float, display_name: str, category_name: str
)

Category is a util class, contains a label, its display name, a float value as score, and the index of the label in the corresponding label file. Typically it's used as the result of classification tasks.

index The index of the label in the corresponding label file.
score The probability score of this label category.
display_name The display name of the label, which may be translated for different locales. For example, a label, "apple", may be translated into Spanish for display purpose, so that the display_name is "manzana".
category_name The label of this category object.

Methods

create_from_pb2

View source

@classmethod
create_from_pb2(
    pb2_obj: _ClassProto
) -> 'Category'

Creates a Category object from the given protobuf object.

to_pb2

View source

to_pb2() -> _ClassProto

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.