我正在使用 theano 和 keras 在 python 中创建一个 CNN。我只想使用 python 在卷积层和池化层之后打印图像大小。我使用 cifar10 数据集,它是 (3, 32, 32),我想知道图像大小减少了多少。
image = X_train[1]
print(image.shape)
使用 Theano 后端。
(3, 32, 32)
最佳答案
如果您使用的是 Sequential() 模型,则可以在插入该层后执行 model.summary()。
如果您使用的是函数式 API,则可以创建一个模型
keras.models.Model(input=my_input_layer,output=my_last_layer).summary()
这有帮助吗?
关于theano - 如何在卷积神经网络中的池化层后打印图像尺寸,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/42684353/