ValueError:检查模型目标时出错:期望dense_4具有形状(None,4)但是具有形状的数组(13252,1)

时间:2017-02-14 20:29:55

标签: machine-learning tensorflow keras keras-layer

嗨,有没有人有任何想法为什么会发生这种错误? 这是错误

ValueError: Error when checking model target: expected dense_4 to have   shape (None, 4) but got array with shape (13252, 1)

以下是代码:

from keras.models import Sequential
from keras.layers import *

model = Sequential()
model.add(Cropping2D(cropping=((0,0), (50,20)), input_shape=(160 ,320, 3))) #(None, 90, 320, 3)
model.add(Lambda(lambda x: x/127.5 - 1.)) 
model.add(Convolution2D(32, 3, 3,)) #(None, 88, 318, 32)
model.add(Activation('relu'))
model.add(Convolution2D(32, 3, 3)) #(None, 86, 316, 32)
model.add(Activation('relu'))
model.add(Flatten()) #(None, 869632)
model.add(Dense(128)) #(None, 128)
model.add(Activation('relu'))
model.add(Dense(4)) #(None, 4)
print(model.summary())

model.compile(loss='mse', optimizer='adam')
model.fit(X, y, validation_split=0.2, batch_size=32, nb_epoch=3, verbose=1)

输入形状为(X):

(13252, 160, 320, 3)

和(y):

(13252,)

1 个答案:

答案 0 :(得分:7)

由于您的网络有四个输出,您的 y 必须是N x 4矩阵,而不是长度为N的向量。要么更改y,要么将最后一层更改为Dense(1)