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