每日Attention学习2——Multi-Scale Convolutional Attention

模块出处

[link] [code] [NIPS 22] SegNeXt: Rethinking Convolutional Attention Design for Semantic Segmentation

---

模块名称

Multi-Scale Convolutional Attention (MSCA)

---

模块作用

多尺度特征提取，更大感受野

---

模块结构

---

模块代码

import torch
import torch.nn as nn

class MSCA(nn.Module):
    def __init__(self, dim):
        super(MSCA, self).__init__()
        self.conv0 = nn.Conv2d(dim, dim, 5, padding=2, groups=dim)
        self