View source on GitHub |
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
)
Args | |
---|---|
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
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. |