GlobalAveragePooling2D和GlobalMaxPooling2D

方如一

已于 2022-06-24 15:23:02 修改

阅读量3.2k

点赞数

分类专栏： Python参数分析文章标签： python tensorflow numpy

于 2022-06-24 10:27:00 首次发布

本文链接：https://blog.csdn.net/Fwuyi/article/details/125439860

版权

Python参数分析专栏收录该内容

8 篇文章

订阅专栏

本文介绍了深度学习中两种常用的全局池化操作：GlobalAveragePooling2D和GlobalMaxPooling2D。GlobalAveragePooling2D通过计算输入数据每个通道的平均值来降低维度，而GlobalMaxPooling2D则是取每个通道的最大值。这两种方法都用于去除图像的宽度和高度维度，保留批次大小和通道数，从而简化网络结构并减少参数数量。示例展示了它们在TensorFlow中的实现和效果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一 GlobalAveragePooling2D 全局平均池化

GlobalAveragePooling2D降低维度。

GlobalAveragePooling2D就是求四维数据（图片）的每个通道值c的平均，最后结果没有了宽（w）高（h）维度，只剩下batchsize与channel两个维度。从形状上看：[B,H,W,C] → [B,C]。

下面有1张图片,shape为[1,2,2,3]，经过GlobalAveragePooling2D得到shape[1,3]。

import tensorflow as tf

x = tf.random.normal((1,2,2,3))
x.shape
#x：  
<tf.Tensor: shape=(1, 2, 2, 3), dtype=float32, 
numpy=array([[[[ 1.0732665 , -0.316923  ,  0.34387082],
               [ 0.85719526,  0.383818  ,  0.0714375 ]],
              [[ 2.07628   ,  1.0945694 ,  0.6519038 ],
               [ 0.00441903,  1.2396817 ,  0.10125866]]]], dtype=float32)>

ax = tf.keras.layers.GlobalAveragePooling2D()(x)
ax.shape
#ax：
<tf.Tensor: shape=(1, 3), dtype=float32, 
numpy=array([[1.0027902 , 0.60028654, 0.29211769]], dtype=float32)>

GlobalAveragePooling2D中的运算是K.mean(input, axis=[1, 2]),可以用np.mean(input, axis=(1, 2))来模拟这个过程。

x = np.array([[[0, 1, 2],
               [3, 4, 5]],
              [[0, 1, 2],
               [3, 4, 5]]])
#x.shape=[2,2,3]
new_xx = np.mean(x, axis=(0, 1))
#new_xx = [2.25, 2.5, 3.5]
#new_xx.shape = [3]
new_x = np.mean(x, axis=0)
#new_x = [[0, 1, 2], [3, 4, 5]]
#new_xx.shape=[2,3]