Toybrick

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

jandriux

注册会员

积分
90
楼主
发表于 2019-7-26 13:37:34    查看: 7278|回复: 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 | 显示全部楼层
想问一下,这个问题该怎么解决
回复

使用道具 举报

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

本版积分规则

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


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