Attention: This MediaPipe Solutions Preview is an early release. Learn more.

mp.tasks.components.containers.Classifications

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

Represents the classification results for a given classifier head.

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

Creates a Classifications object from the given protobuf object.

to_pb2

View source

Generates a Classifications protobuf object.

__eq__

head_name None