如何在卷积神经网络中合并图层后打印图像大小

Question

我正在使用theano和keras在python中创建CNN。我想只使用python在Convolutional layer和Pooling layer之后打印图像大小。我使用的是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()

有帮助吗？