View source on GitHub
|
TF Lite Metadata Writer API.
This module provides interfaces for writing metadata for common model types supported by the task library, such as:
- Image classification
- Object detection
- Image segmentation
- (Bert) Natural language classification
- Audio classification
It is provided as part of the tflite-support package:
pip install tflite-support
Learn more about this API in the metadata writer tutorial.
Modules
audio_classifier module: Writes metadata and label file to the audio classifier models.
bert_nl_classifier module: Writes metadata and label file to the Bert NL classifier models.
image_classifier module: Writes metadata and label file to the image classifier models.
image_segmenter module: Writes metadata and label file to the image segmenter models.
metadata_info module: Helper classes for common model metadata information.
nl_classifier module: Writes metadata and label file to the NL classifier models.
object_detector module: Writes metadata and label file to the object detector models.
writer_utils module: Helper methods for writing metadata into TFLite models.