1

我正在为 MNIST 任务微调 VGG19 模型。MNIST中的图像是(28,28,1)，是一个通道。但是VGG想要输入的是(?,?,3)，也就是三个通道。

所以，我的方法是在所有 VGG 层之前再添加一个 Conv2D 层，将 (28,28,1) 数据更改为 (28,28,3)，这是我的代码：

inputs = Input(shape=(28,28,1))
x = Conv2D(3,kernel_size=(1,1),activation='relu')(inputs)
print(x.shape)
# out: (?, 28, 28, 3)

现在我的输入形状是正确的（我认为）。

这是我的整个模型： # 更改输入形状：inputs = Input(shape=(28,28,1)) x = Conv2D(3,kernel_size=(1,1),activation='relu')(inputs)

# add POOL and FC layers:
x = base_model(x)
x = GlobalMaxPooling2D()(x)
x = Dense(1024,activation='relu')(x)
predictions = Dense(10,activation='softmax')(x)

model = Model(inputs=inputs, outputs=predictions)

# freeze the base_model:
for layer in base_model.layers:
    layer.trainable = False

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

我得到了：

InvalidArgumentError: Negative dimension size caused by subtracting 2 from 1 for 'vgg19_10/block5_pool/MaxPool' (op: 'MaxPool') with input shapes: [?,1,1,512].

我已经搜索了问题，一种解决方案是添加

from keras import backend as K
K.set_image_dim_ordering('th')

但这对我不起作用。

我的代码有什么问题？

4

1 回答 1

1

当我们进行分类时，我们应该在输出上使用 softmax 激活。

将最后一层的激活更改为 softmax

从

predictions = Dense(10,activation='relu')(x)

至

predictions = Dense(10,activation='softmax')(x)

导致错误的第二个错误与您的输入大小有关。根据keras vgg19 ，您的最小图像尺寸应不小于 48。

inputs = Input(shape=(48,48,1))
于 2018-07-31T08:56:39.223 回答