检查输入时出错:预期conv2d_6_input具有4个维度,但数组的形状为(270,50,50)

时间:2019-07-26 14:29:10

标签: python tensorflow keras

图片(50x50)

model =Sequential()
model.add(Conv2D(64,3,3,input_shape=(50,50,1)))
model.add(Activation('relu'))
model.add(MaxPooling2D(pool_size=(2,2)))

model.add(Conv2D(64,3,3,input_shape=(50,50,1)))
model.add(Activation('relu'))
model.add(MaxPooling2D(pool_size=(2,2)))

model.add(Flatten())
model.add(Dense(64))

model.add(Dense(5))
model.add(Activation('softmax'))



model.fit(X_train,y_train,epochs=5)

错误:检查输入时出错:预期conv2d_6_input具有4维,但数组的形状为(270,50,50)

1 个答案:

答案 0 :(得分:0)

Conv2D需要4个尺寸,因为它需要一个彩色通道(图像可以是灰度,RGB等)

您需要使用np.expand_dims

扩展尺寸以说明该渠道