Introducing LiteRT: Google's high-performance runtime for on-device AI, formerly known as TensorFlow Lite.
Learn more
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
)
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
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 protobuf object to pass to the C++ layer.
__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.
|
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.
Last updated 2024-05-08 UTC.
[{
"type": "thumb-down",
"id": "missingTheInformationINeed",
"label":"Missing the information I need"
},{
"type": "thumb-down",
"id": "tooComplicatedTooManySteps",
"label":"Too complicated / too many steps"
},{
"type": "thumb-down",
"id": "outOfDate",
"label":"Out of date"
},{
"type": "thumb-down",
"id": "samplesCodeIssue",
"label":"Samples / code issue"
},{
"type": "thumb-down",
"id": "otherDown",
"label":"Other"
}]
[{
"type": "thumb-up",
"id": "easyToUnderstand",
"label":"Easy to understand"
},{
"type": "thumb-up",
"id": "solvedMyProblem",
"label":"Solved my problem"
},{
"type": "thumb-up",
"id": "otherUp",
"label":"Other"
}]
{"lastModified": "Last updated 2024-05-08 UTC."}
[[["Easy to understand","easyToUnderstand","thumb-up"],["Solved my problem","solvedMyProblem","thumb-up"],["Other","otherUp","thumb-up"]],[["Missing the information I need","missingTheInformationINeed","thumb-down"],["Too complicated / too many steps","tooComplicatedTooManySteps","thumb-down"],["Out of date","outOfDate","thumb-down"],["Samples / code issue","samplesCodeIssue","thumb-down"],["Other","otherDown","thumb-down"]],["Last updated 2024-05-08 UTC."],[],[]]