tflite_support.metadata_writers.metadata_info.CategoryTensorMd

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

A container for the category tensor metadata information.

Inherits From: TensorMd

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.