YoloV8模块添加（新手初探，请大佬轻喷）

一.路径查找

修改模型的前提是要找对文件，路径错了，修改也就失去了意义。

修改前，请先正确安装yolo，强烈建议按照上一个博文进行安装。

v8的模块修改主要集中在：?:/ultralytics/ultralytics/nn/modules/

如果忘记自己的包在哪里，推荐使用   pip list   查看

二.注意力模块添加，以SE-Net为例

class se_block(nn.Module):   #SEnet
    def __init__(self, channel, reduction=16):
        super(se_block, self).__init__()
        self.avg_pool = nn.AdaptiveAvgPool2d(1)
        self.fc = nn.Sequential(
            nn.Linear(channel, channel // reduction, bias=False),
            nn.ReLU(inplace=True),
            nn.Linear(channel // reduction, channel, bias=False),
            nn.Sigmoid()
        )

    def forward(self, x):
        b, c, _, _ = x.size()
        y = self.avg_pool(x).view(b, c)
        y = self.fc(y).view(b, c, 1, 1)
        return x * y.expand_as(x)

1.添加类

将上述代码添加到在ultralytics/nn/modules/block(conv也行).py文件内。

2.引用

修改ultralytics/nn/modules/__init__.py文件，添加模块的引用。是ultralytics/nn/modules/__init__.py哦，别添加到ultralytics/nn/__init__.py了。

这里要注意的是：代码添加到那个文件就把引用添加到哪一个块中。前面我们添加的是block，这里引用也要添加到block中。

这个添加到前面后面中间都是一样的，有就行，看你的喜好。中间用注释画个坤坤都行。

三.调用

修改？:/ultralytics/ultralytics/nn/tasks.py文件。我这里使用的是8.版本，只需要添加这一个便可以。

四.修改Yaml文件

我这里是把se添加到SPFF前，这是第9层，后面的部分也要做修改，很简单，大于9的数字都加1就行。

最后执行指令，打印出的模型结构是有SE block的

YoloV8模块添加（新手初探，请大佬轻喷）