Keras用於多位數識別

Question

我想通過Keras（theano後端）中的一些練習來了解CNNs。我無法適應下面的模型（錯誤：AttributeError：'Convolution2D'對象沒有'get_shape'屬性）。該數據集是來自MNIST數據的圖像（28 * 28），最多連接5個圖像。所以輸入形狀應該是1,28,140（灰度= 1，高度= 28，寬度= 28 * 5）

目標是預測數字序列。謝謝！！

batch_size = 128 
nb_classes = 10 
nb_epoch = 2 

img_rows =28 
img_cols=140 
img_channels = 1 

model_input=(img_channels, img_rows, img_cols) 

x = Convolution2D(32, 3, 3, border_mode='same')(model_input) 
x = Activation('relu')(x) 
x = Convolution2D(32, 3, 3)(x) 
x = Activation('relu')(x) 
x = MaxPooling2D(pool_size=(2, 2))(x) 
x = Dropout(0.25)(x) 
conv_out = Flatten()(x) 

x1 = Dense(nb_classes, activation='softmax')(conv_out) 
x2 = Dense(nb_classes, activation='softmax')(conv_out) 
x3 = Dense(nb_classes, activation='softmax')(conv_out) 
x4 = Dense(nb_classes, activation='softmax')(conv_out) 
x5 = Dense(nb_classes, activation='softmax')(conv_out) 

lst = [x1, x2, x3, x4, x5] 

model = Sequential(input=model_input, output=lst) 

model.compile(loss='categorical_crossentropy', 
optimizer='adam', 
metrics=['accuracy']) 

model.fit(dataset, data_labels, batch_size=batch_size, nb_epoch=nb_epoch, verbose=1) 

Answer 1

問題出在輸入層。執行以下更改：

model_input=Input(shape=(img_channels, img_rows, img_cols)) 

提供您的image_dim_ordering是th。從keras.layers導入Input層。

我還注意到有多個輸出。所以你需要使用函數模型而不是Sequential。從keras.models

model = Model(input=model_input, output=lst) 

進口Model：只需將其更改爲。