Torch版本主要分为四部分,ResNet、ResNeSt、split-attention和ablation

split-attention模块也就是下面的图:


上图对应论文中的:

ResNet就是基本的resnet的编码:
class Bottleneck(nn.Module):
就是对ResNet的bottleneck的编写,需要说的是,全局平均池化没有采用之前的网络采取对池化函数进行编码的方式,如下所:

而是采用的一个函数:

与此同时,split-attention也是放在了bottleneck函数中,如下:

本文深入探讨Torch版本的四大核心部分:ResNet、ResNeSt、split-attention及ablation研究。详细分析了ResNet编码、bottleneck设计与split-attention模块的集成方式,同时介绍了ResNeSt网络结构及其预训练过程,最后讨论了ablation实验的设计与结果。
ResNeSt Split-Attention Networks&spm=1001.2101.3001.5002&articleId=105786674&d=1&t=3&u=554d70c440a2422f836a265582f59236)
3882

被折叠的 条评论
为什么被折叠?



