小编Nic*_*lla的帖子

使用 keras ImageGenerator 训练多输入模型

我建立了一个模型,该模型由两个分支组成,然后合并为一个分支。对于模型的训练,我想使用 ImageGenerator 来增强图像数据,但不知道如何为混合输入类型工作。有人知道如何在 keras 中处理这个问题吗?任何帮助将不胜感激!

最好的,尼克

模型一个分支将图像作为输入:

img_model = Sequential()
img_model.add(Convolution2D( 4, 9,9, border_mode='valid', input_shape=(1, 120, 160)))
img_model.add(Activation('relu'))
img_model.add(MaxPooling2D(pool_size=(2, 2)))
img_model.add(Dropout(0.5))
img_model.add(Flatten()) 
Run Code Online (Sandbox Code Playgroud)

第二个分支将辅助数据作为输入:

aux_model = Sequential()
aux_model.add(Dense(3, input_dim=3))
Run Code Online (Sandbox Code Playgroud)

然后将它们合并到最终模型中

model = Sequential()
model.add(Merge([img_model, aux_model], mode='concat'))
model.add(Dropout(0.5))
model.add(Dense(5))
model.add(Activation('softmax'))
model.compile(loss='categorical_crossentropy', optimizer='adadelta', metrics=['accuracy']) 
Run Code Online (Sandbox Code Playgroud)

训练/问题: 我尝试执行以下操作,但显然失败了:

datagen = ImageDataGenerator(
            featurewise_center=False,  # set input mean to 0 over the dataset
            samplewise_center=False,  # set each sample mean to 0
            featurewise_std_normalization=False,  # divide inputs by std of …
Run Code Online (Sandbox Code Playgroud)

python neural-network deep-learning keras

5
推荐指数
1
解决办法
2825
查看次数

标签 统计

deep-learning ×1

keras ×1

neural-network ×1

python ×1