Module: tflite_support.metadata_writers

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.