Attend the Women in ML Symposium on December 7 Register now

Module: tflite_support.metadata_writers

Stay organized with collections Save and categorize content based on your preferences.

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.