Pytorch的adaptive_avg_pool2d算法

时间:2018-10-15 01:31:27

标签: pytorch pooling

任何人都知道pytorchadaptive_avg_pool2d的算法,例如

adaptive_avg_pool2d(image,[14,14])

这样的问题:     我想在keras神经网络中做同样的事情,对于任何给定输入,都想获得14 * 14的输出。有什么建议吗?

1 个答案:

答案 0 :(得分:1)

我认为这在Keras中不存在。您可以获取输入的维度,然后除以14得到所需的pool_size。

例如,如果您的输入为28x28,则可以使用:

keras.layers.AveragePooling2D(pool_size=(2, 2), strides=None, padding='valid', data_format=None)