deep-learning - keras 密集输入层

Question

我主要是 caffe 用户，我想知道如何在全连接层中输入我的彩色图像，然后将其展平（我知道这不是最好的解决方案，但我需要它）。这是一个不起作用的示例：

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

有什么建议吗？我想我必须设置输入层，然后将其展平，但我不知道如何。

提前致谢。

Answer 1

刚刚找到解决方案...

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

不过很明显。:(