mp.tasks.components.containers.Classifications

Represents the classification results for a given classifier head.

mp.tasks.components.containers.Classifications(
    categories: List[mp.tasks.components.containers.Category],
    head_index: int,
    head_name: Optional[str] = None
)

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

Methods

create_from_pb2

View source

@classmethod
create_from_pb2(
    pb2_obj: _ClassificationsProto
) -> 'Classifications'

Creates a Classifications object from the given protobuf object.

to_pb2

View source

to_pb2() -> _ClassificationsProto

Generates a Classifications protobuf object.

__eq__

__eq__(
    other
)

Class Variables
head_name	None