tflite_support.metadata_writers.metadata_info.CategoryTensorMd

A container for the category tensor metadata information.

Inherits From: TensorMd

Main aliases

tflite_support.metadata_writers.audio_classifier.metadata_info.CategoryTensorMd, tflite_support.metadata_writers.audio_classifier.metadata_writer.metadata_info.CategoryTensorMd, tflite_support.metadata_writers.bert_nl_classifier.metadata_info.CategoryTensorMd, tflite_support.metadata_writers.bert_nl_classifier.metadata_writer.metadata_info.CategoryTensorMd, tflite_support.metadata_writers.image_classifier.metadata_info.CategoryTensorMd, tflite_support.metadata_writers.image_classifier.metadata_writer.metadata_info.CategoryTensorMd, tflite_support.metadata_writers.image_segmenter.metadata_info.CategoryTensorMd, tflite_support.metadata_writers.image_segmenter.metadata_writer.metadata_info.CategoryTensorMd, tflite_support.metadata_writers.nl_classifier.metadata_info.CategoryTensorMd, tflite_support.metadata_writers.nl_classifier.metadata_writer.metadata_info.CategoryTensorMd, tflite_support.metadata_writers.object_detector.metadata_info.CategoryTensorMd, tflite_support.metadata_writers.object_detector.metadata_writer.metadata_info.CategoryTensorMd

 
tflite_support.metadata_writers.metadata_info.CategoryTensorMd(
    name: Optional[str] = None,
    description: Optional[str] = None,
    label_files: Optional[List[LabelFileMd]] = None
)

name name of the tensor.
description description of what the tensor is.
label_files information of the label files 1 in the category tensor.

Methods

create_metadata

View source

create_metadata() -> tflite_support.metadata_schema_py_generated.TensorMetadataT

Creates the input tensor metadata based on the information.

Returns
A Flatbuffers Python object of the input metadata.