Module: tflite_support.metadata_writers.image_classifier

Writes metadata and label file to the image classifier models.

Modules

metadata_info module: Helper classes for common model metadata information.

metadata_writer module: Helper class to write metadata into TFLite models.

writer_utils module: Helper methods for writing metadata into TFLite models.

Classes

class MetadataWriter: Writes metadata into an image classifier.

INPUT_DESCRIPTION 'Input image to be classified.'
INPUT_NAME 'image'
MODEL_DESCRIPTION ('Identify the most prominent object in the image from a known set of ' 'categories.')
OUTPUT_DESCRIPTION 'Probabilities of the labels respectively.'
OUTPUT_NAME 'probability'