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
以后真的要好好研究基础和经典了。