在pytorch的PSPNet中
在这里插入图片描述
网络构建有用到nn.AdaptiveAvgPool2d(pool_size)
在这里插入图片描述
其中参数pool_size为给定输出特征图的大小,其中通道数前后不发生变化。

  • 输入(b,c,h,w)\color{red}{(b,c,h,w)}bchw
  • 输出(b,c,poolsize,poolsize)\color{red}{(b,c,pool_size,pool_size)}bcpoolsizepoolsize

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

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐