Introducing LiteRT : Google's high-performance runtime for on-device AI, formerly known as TensorFlow Lite.
Learn more
Send feedback
tflite_support.task.vision.ImageClassifier
Stay organized with collections
Save and categorize content based on your preferences.
Class that performs classification on images.
tflite_support . task . vision . ImageClassifier (
options : tflite_support . task . vision . ImageClassifierOptions
,
classifier : _CppImageClassifier
) -> None
Methods
classify
View source
classify (
image : tflite_support . task . vision . TensorImage
,
bounding_box : Optional [ tflite_support . task . processor . BoundingBox
] = None
) -> tflite_support . task . processor . ClassificationResult
Performs classification on the provided TensorImage.
Args
image
Tensor image, used to extract the feature vectors.
bounding_box
Bounding box, optional. If set, performed feature vector
extraction only on the provided region of interest. Note that the region
of interest is not clamped, so this method will fail if the region is
out of bounds of the input image.
Returns
classification result.
Raises
ValueError
If any of the input arguments is invalid.
RuntimeError
If failed to run classification.
create_from_file
View source
@classmethod
create_from_file (
file_path : str
) -> 'ImageClassifier'
Creates the ImageClassifier
object from a TensorFlow Lite model.
Args
file_path
Path to the model.
Returns
ImageClassifier
object that's created from the model file.
Raises
ValueError
If failed to create ImageClassifier
object from the
provided file such as invalid file.
RuntimeError
If other types of error occurred.
create_from_options
View source
@classmethod
create_from_options (
options : tflite_support . task . vision . ImageClassifierOptions
) -> 'ImageClassifier'
Creates the ImageClassifier
object from image classifier options.
Args
options
Options for the image classifier task.
Returns
ImageClassifier
object that's created from options
.
Raises
ValueError
If failed to create ImageClassifier
object from
ImageClassifierOptions
such as missing the model.
RuntimeError
If other types of error occurred.
Send feedback
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"
}]
Need to tell us more?
{"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."],[],[]]