Toybrick

模型转换新人贴

chenli

注册会员

积分
133
楼主
发表于 2020-5-15 18:02:02    查看: 7166|回复: 1 | [复制链接]    打印 | 只看该作者
这两天转换模型始终报错,经我个人推敲应该是申请变量出错,其实好几种方法在pc都能用但转rknn就报错,而且示例代码给的是三通道的我用的是灰度图,请大神帮我看一下这段代码该怎么写(我相信做过深度学习的都知道着带码是什么意思,我就不废话了上带码!)
inputs = tf.placeholder(tf.float32,[None,128,128],name='inputs')  # 神经网络输入数据(1*4096)
inputs_reshape = tf.reshape(inputs, [-1, 128, 128, 1])  # 改变经网络输入数据形状(164*64的矩阵)
labels = tf.placeholder(tf.float32, [None,32],name='labels')  # 神经网络输出数据的形状(1*3)
kp = tf.placeholder_with_default(1.0, shape=(), name="keep_prob"
请大神指教总是inputlayer报错


回复

使用道具 举报

jefferyzhang

版主

积分
13580
沙发
发表于 2020-5-15 18:07:46 | 只看该作者
你如果自己写模型的话,为啥不用keras写法,还在用这种被淘汰的写法呢。。。
回复

使用道具 举报

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

本版积分规则

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


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