小编Col*_*ick的帖子

TFLiteterpreter.get_input_details 中的“索引”指的是什么?

我正在努力使用 Keras/TensorFlow 模型进行一些推理,但文档似乎有点稀疏,因此我尝试尽可能多地学习和记录,而不仅仅是依赖复制的代码示例。示例包括这一行:

interpreter.get_input_details()

它返回一个包含单个字典的列表,其中包含以下信息:

[{'name': 'conv2d_input', 'index': 8, 'shape': array([ 1, 28, 28, 1]), 'dtype': <class 'numpy.float32'>, 'quantization': (0.0, 0)}]

get_output_details返回非常相似的东西。其中大部分内容是不言自明的,但我不清楚该索引指的是什么。我发现的示例表明 set_tensor 和 get_tensor 方法需要该数字。到目前为止我发现的只是某处的一条评论说这指的是张量索引,这对我来说似乎是同义反复!

提前致谢。

keras tensorflow tensorflow-lite

8
推荐指数
1
解决办法
3082
查看次数

标签 统计

keras ×1

tensorflow ×1

tensorflow-lite ×1