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

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

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