基于tensorflow的mnist模型转rknn，input_size_list 不等于inputs

在mnist.py中训练模型时，定义为：
x = tf.compat.v1.placeholder(tf.float32, [None, 784], name="read_image")



在转换的时候：
rknn.load_tensorflow(tf_pb='mnist.pb',
                         inputs=['read_image'],
                         outputs=['final_result'],
                         input_size_list=[[28,28]])

运行脚本的时候，报错：
length of input_size_list is not equal to inputs，请问input_size_list需要做怎样的修改。

想问一下，这个问题该怎么解决

jandriux 发表于 2019-9-16 16:52
想问一下，这个问题该怎么解决

应该是[28, 28, 1]吧

对于MNIST还有问题的，我们已经让同事做了一个mnist转换和量化的教程了，
参考：http://t.rock-chips.com/forum.ph ... &extra=#pid4127