pytorch nn.AdaptiveAvgPool2d(1)

在pytorch的PSPNet中
pytorch nn.AdaptiveAvgPool2d(1)_第1张图片
网络构建有用到nn.AdaptiveAvgPool2d(pool_size)
pytorch nn.AdaptiveAvgPool2d(1)_第2张图片
其中参数pool_size为给定输出特征图的大小,其中通道数前后不发生变化。

  • 输入 ( b , c , h , w ) \color{red}{(b,c,h,w)} bchw
  • 输出 ( b , c , p o o l s i z e , p o o l s i z e ) \color{red}{(b,c,pool_size,pool_size)} bcpoolsizepoolsize

全局信息采用如下公式:
在这里插入图片描述

你可能感兴趣的:(深度学习,pytorch,人工智能,python,神经网络,计算机视觉)