theano - 如何在卷积神经网络中的池化层后打印图像尺寸

Question

我正在使用 theano 和 keras 在 python 中创建一个 CNN。我只想使用 python 在卷积层和池化层之后打印图像大小。我使用 cifar10 数据集，它是 (3, 32, 32)，我想知道图像大小减少了多少。

image = X_train[1]
print(image.shape)

使用 Theano 后端。

(3, 32, 32)

Answer 1

如果您使用的是 Sequential() 模型，则可以在插入该层后执行 model.summary()。

如果您使用的是函数式 API，则可以创建一个模型

keras.models.Model(input=my_input_layer,output=my_last_layer).summary()

这有帮助吗？