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
The array of predicted categories, usually sorted by descending
scores (e.g. from high to low probability).
The index of the classifier head these categories refer to. This
is useful for multi-head models.
Methods
create_from_pb2
@classmethodcreate_from_pb2 ( pb2_obj : _ClassificationsProto ) -> 'Classifications'
Creates a Classifications
object from the given protobuf object.
to_pb2
to_pb2
()
->
_ClassificationsProto
Generates a protobuf object to pass to the C++ layer.
__eq__
__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.
