tflite_support.metadata_writers.metadata_info.ClassificationTensorMd

A container for the classification tensor metadata information.

Inherits From: TensorMd

name name of the tensor.
description description of what the tensor is.
label_files information of the label files 1 in the classification tensor.
tensor_type data type of the tensor.
score_calibration_md information of the score calibration files operation 2 in the classification tensor.
tensor_name name of the corresponding tensor 3 in the TFLite model. It is used to locate the corresponding classification tensor and decide the order of the tensor metadata 4 when populating model metadata.

label_files information of the label files 1 in the classification tensor.
score_calibration_md information of the score calibration operation 2 in the classification tensor.

Methods

create_metadata

View source

Creates the classification tensor metadata based on the information.