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')) 

Это было очевидно, хотя. :(