我正在以tensorflow backhend的形式在keras中运行以下代码片段。
网络架构:
model = Sequential()
model.add(Convolution2D(32, 3, 3, border_mode="same",input_shape=(numChannels,imgRows, imgCols)))
model.add(Activation("relu"))
model.fit(trainData, trainLabels, batch_size=batch_size,nb_epoch=20, verbose=verbose)
我收到一个错误,因为:
ValueError:检查模型输入时的错误:期望convolution2d_input_1有4个维,但得到了形状为(89696,1024)的数组
怎么解决这个问题?提前谢谢!!
发布于 2017-12-01 12:03:45
trainData
格式不正确,它应该具有形状(numImages, numChannels, imgRows, imgCols)
。
https://stackoverflow.com/questions/47599888
复制