目录
1. DoubleAttention模块
2. SwinTransformer + DoubleAttention
3. 完整代码
Tips:融入模块后的网络经过测试,可以直接使用,设置好输入和输出的图片维度即可
1. DoubleAttention模块
DoubleAttention
是一种用于计算机视觉任务的注意力机制,旨在通过双重注意力机制增强特征表示。它通常用于图像分类、目标检测和语义分割等任务中,以提高模型对重要特征的捕捉能力。
核心思想
DoubleAttention
模块的核心思想是通过两种不同的注意力机制(通常是 通道注意力 和 空间注意力)来增强特征表示:
-
通道注意力