keras密集输入层

时间:2015-12-18 09:44:19

标签: deep-learning keras

我主要是一个咖啡用户,我想知道如何在完全连接的层中输入我的彩色图像然后压平它(我知道这不是最好的解决方案,但我需要它)。 这是一个不起作用的样本:

model = Sequential()
model.add(Dense(1000, input_shape=(img_channels, img_rows, img_cols)))
model.add(Activation('relu'))

有什么建议吗?我想我必须设置输入层然后压扁它,但我不知道如何。

提前致谢。

1 个答案:

答案 0 :(得分:4)

刚刚找到解决方案......

model = Sequential()
model.add(Flatten(input_shape=(img_channels, img_rows, img_cols)))
model.add(Dense(1000))
model.add(Activation('relu'))

但显而易见。 :(