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的输入就是单个样本。

0
0

Google老师亲授 TensorFlow2.0 入门到进阶

Tensorflow2.0实战—以实战促理论的方式学习深度学习

1849 学习 · 896 问题

查看课程