keras CNN 如何将训练彩色图片的网络改成灰度图片的网络

就这个问题困惑了一天,终于在一天要结束的时候解决了,一整天都在转换类型,比如我试了这个:

train_data=train_data.reshape(train_data.shape[0],-1)/255 #Normalize

 test_data=test_data.reshape(test_data.shape[0],-1)/255 

还有这个:
train_data=train_data.reshape(len(train_data),-1)

test_data=test_data.reshape(len(test_data),-1)

都不解决问题,查了各种博客未见此问题,终于在GitHub上找到keras实现mnist 的代码,眼高手低说的就是我了,实践还是要一点一点来。

最终解决问题的代码是:

train_data =train_data.reshape(train_data.shape[0], 50, 50, 1)
test_data = test_data.reshape(test_data.shape[0], 50, 50, 1)

train_data = train_data.astype('float32')
test_data = test_data.astype('float32')

train_data/=255
test_data/=255

附上经典的keras实现mnist链接:https://github.com/yashk2810/MNIST-Keras/blob/master/Notebook/MNIST_keras_CNN_99.43%25.ipynb

以后真的要好好研究基础和经典了。

 

  • 5
    点赞
  • 14
    收藏
    觉得还不错? 一键收藏
  • 5
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值