tflite_support.task.processor.Classifications

Stay organized with collections Save and categorize content based on your preferences.

List of predicted classes (aka labels) for a given classifier head.

tflite_support.task.processor.Classifications(
    categories: List[tflite_support.task.processor.Category],
    head_index: int,
    head_name: str
)

categories The array of predicted categories, usually sorted by descending scores (e.g. from high to low probability).
head_index The index of the classifier head these categories refer to. This is useful for multi-head models.
head_name The name of the classifier head, which is the corresponding tensor metadata.

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.