Toybrick

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

jandriux

注册会员

积分
90
发表于 2019-7-26 13:37:34    查看: 7682|回复: 3 | [复制链接]    打印 | 显示全部楼层
在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

注册会员

积分
90
 楼主| 发表于 2019-9-16 16:52:37 | 显示全部楼层
想问一下,这个问题该怎么解决
回复

使用道具 举报

troy

版主

积分
2370
发表于 2019-9-17 08:43:19 | 显示全部楼层
jandriux 发表于 2019-9-16 16:52
想问一下,这个问题该怎么解决

应该是[28, 28, 1]吧
回复

使用道具 举报

jefferyzhang

版主

积分
13111
发表于 2019-9-20 16:41:15 | 显示全部楼层
对于MNIST还有问题的,我们已经让同事做了一个mnist转换和量化的教程了,
参考:http://t.rock-chips.com/forum.ph ... &extra=#pid4127
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表