keras 卷积神经网络 - 输出形状

Question

请原谅我的无知，因为我是这个地区的新手。我试图从我的神经网络中获得正确的输出形状，它有 3 个 Conv2D 层，然后是 2 个密集层。我输入的shape是(140, 140, 4)，是4张灰度图。当我适合 1 个输入时，我期望输出 (1, 4) 但我在这里得到 (14, 14, 4) 的形状。我在这里做错了什么？非常感谢您的提前帮助！

meta_layers = [Conv2D, Conv2D, Conv2D, Dense, Dense]
meta_inits = ['lecun_uniform'] * 5
meta_nodes = [32, 64, 64, 512, 4]
meta_filter = [(8,8), (4,4), (3,3), None, None]
meta_strides = [(4,4), (2,2), (1,1), None, None]
meta_activations = ['relu'] * 5
meta_loss = "mean_squared_error"
meta_optimizer=RMSprop(lr=0.00025, rho=0.9, epsilon=1e-06)
meta_n_samples = 1000
meta_epsilon = 1.0;
    meta = Sequential()
    meta.add(self.meta_layers[0](self.meta_nodes[0], init=self.meta_inits[0], input_shape=(140, 140, 4), kernel_size=self.meta_filters[0], strides=self.meta_strides[0]))
    meta.add(Activation(self.meta_activations[0]))
    for layer, init, node, activation, kernel, stride in list(zip(self.meta_layers, self.meta_inits, self.meta_nodes, self.meta_activations, self.meta_filters, self.meta_strides))[1:]:
        if(layer == Conv2D):
            meta.add(layer(node, init = init, kernel_size = kernel, strides = stride))
            meta.add(Activation(activation))
        elif(layer == Dense):
            meta.add(layer(node, init=init))
            meta.add(Activation(activation))
        print("meta node: " + str(node))
    meta.compile(loss=self.meta_loss, optimizer=self.meta_optimizer)

Answer 1

您的问题在于，在版本 >= 2.0 的 Keras 中，Dense 层应用于输入的最后一个通道（您可能会读到它）。所以如果你申请：

Dense(512)

到形状为 (14, 14, 64) 的 Conv2D 层，你将得到形状为 (14, 14, 512) 的输出，然后将 Dense(4) 应用到它会得到形状为 (14, 14, 4)。你可以调用model.summary()方法来确认我的话。

为了解决这个问题，您需要应用以下层之一：GlobalMaxPooling2D、GlobalAveragePooling2D 或 Flatten 到最后一个卷积层的输出，以压缩您的输出仅为二维（形状为 (batch_size, features).

keras 卷积神经网络 - 输出形状

keras convolutional nerural network - output shape

machine-learning

neural-network

conv-neural-network

keras

keras-layer