Toybrick

标题: 基于tensorflow的mnist模型转rknn，input_size_list 不等于inputs [打印本页]

作者: jandriux 时间: 2019-7-26 13:37
标题: 基于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
想问一下，这个问题该怎么解决

作者: troy 时间: 2019-9-17 08:43

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

应该是[28, 28, 1]吧

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

欢迎光临 Toybrick (https://t.rock-chips.com/)