pytorch中global average pooling怎么写

最新推荐文章于 2025-03-13 15:30:16 发布

hahajing369

最新推荐文章于 2025-03-13 15:30:16 发布

阅读量1w

点赞数 11

本文链接：https://blog.csdn.net/jingtingxu369/article/details/110659631

版权

写法1
假如特征图是512，7，7，要变成512，1，1，使用GAP
avgpool = nn.AvgPool2d(7, stride=1)   #AvgPool2d(kernel_size=7, stride=1, padding=0)，kernel大小为7*7

写法2，feature_map大小torch.Size([8, 21, 448, 448])，经过adaptive_avg_pool之后结果是torch.Size([8, 21, 1, 1])
avgpool = F.adaptive_avg_pool2d(feature_map, (1,1))

写法3，feature_map大小torch.Size([8, 21, 448, 448])，经过adaptive_avg_pool之后结果是torch.Size([8, 21, 1, 1])
avgpool = F.avg_pool2d(x, kernel_size=(448, 448), padding=0)