首页 文章

keras密集输入层

提问于
浏览
3

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

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

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

提前致谢 .

1 回答

  • 4

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

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

    但很明显 . :(

相关问题