是否有等效于C ++的tf.lite.Interpreter.get_input_details？

Question

在TensorFlow lite的Python API中，有一些方法可以检索有关输入和输出张量的详细信息，称为tf.lite.Interpreter.get_input_details和tf.lite.Interpreter.get_output_details。我在这些字典中输入了“量化”之后，其中每个均包含平均值和标准差的值。

在其他平台上部署这些模型时，我使用C ++ API来设置解释器。要为模型准备输入，我还需要均值和标准差。

我无法在C ++中找到这些函数的等效项。它们存在吗？

Answer 1

您可以从`TfLiteTensor结构中获取量化参数，例如...

// Get the input tensor indices.
const std::vector<int>& inputs = interpreter->inputs();
// Get the tensor object of the 1st input.
TfLiteTensor* tensor = interpreter->tensor(input[0]);
const TfLiteQuantizationParams& params = tensor->param

Answer 2

您也可以使用PrintInterpreterState()来获取相同的信息，实际上是非常麻烦的：

tflite::PrintInterpreterState(interpreter.get());