tflite_support.task.processor.Category

View source on GitHub

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.

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