魔改YOLOv5算法系列（1）——注意力机制添加

小伙伴们，该如何给YOLOv5算法网络添加注意力：

主要改这三个文件：分别是 common.py、yolo.py以及以及配置文件.yaml。

1.打开common.py，复制粘贴注意力模块

 例如:卷积注意力CBAM

CBAM代码段：

class ChannelAttention(nn.Module):
    def __init__(self, in_planes, ratio=16):
        super(ChannelAttention, self).__init__()
        self.avg_pool = nn.AdaptiveAvgPool2d(1)
        self.max_pool = nn.AdaptiveMaxPool2d(1)
 
        self.f1 = nn.Conv2d(in_planes, in_planes // ratio, 1, bias=False)
        self.relu = nn.ReLU()
        self.f2 = nn.Conv2d(in_planes // ratio, in_planes, 1, bias=False)
        # 写法二,亦可使用顺序容器
        # self.sharedMLP = nn.Sequential(
        # nn.Conv2d(in_planes, in_planes // ratio, 1, bias=False), nn.ReLU(),
        #