input_details的shape
来源:9-6 tflite保存与解释与量化
wxz123
2019-12-31
input_details = interpreter.get_input_details()
output_details = interpreter.get_output_details()
print(input_details)
print(output_details)
[{'name': 'x', 'index': 10, 'shape': array([ 1, 28, 28], dtype=int32), 'dtype': <class 'numpy.float32'>, 'quantization': (0.0, 0)}]
[{'name': 'Identity', 'index': 0, 'shape': array([ 1, 10], dtype=int32), 'dtype': <class 'numpy.float32'>, 'quantization': (0.0, 0)}]
老师这里的input_details的 'shape’觉得不是很合理,怎么是[ 1, 28, 28],不应该是[-1,28,28]吗?
写回答
1回答
-
正十七
2020-01-02
同学你好,这里应该是1, 因为tflite是给inference用的,而inference的输入就是单个样本。
00
相似问题
关于decoder中的QKV问题
回答 2
查看网络各个层输出的形状
回答 1