tensorflow CNN 坑

在测试数据集的准确度的时候

出现报错,Failed to convert object of to Tensor. Contents...

然后  报错的是这一行

y =mnist_inference_lenet.inference(x,regularizer=None,train=False)

经查,是inference中的卷积层到全链接层的转换出错,第一维最好用-1代替,保证不会出错

之前训练的时候并没有报错,但是训练的测试的数据大小并不一样,导致维度错误。

# pool_shape = pool2.get_shape().as_list()
# nodes = pool_shape[1]*pool_shape[2]*pool_shape[3]
#
# reshaped= tf.reshape(pool2,[pool_shape[0],nodes])
pool_shape = pool2.get_shape().as_list()
nodes = pool_shape[1] * pool_shape[2] * pool_shape[3]
reshaped = tf.reshape(pool2, [-1, nodes])
另外,数据的维度一定要对应,包括X,Y都要正确。
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值