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

__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.