yolov8+SENet

退即是进

已于 2024-01-19 10:24:16 修改

阅读量880

点赞数 11

文章标签： YOLO

于 2024-01-17 16:25:28 首次发布

本文链接：https://blog.csdn.net/weixin_42875510/article/details/135653310

版权

自学备用

1 在路径打开 yolov8中的conv.py文件，路径如下：ultralytics/ultralytics/nn/modules/conv.py

添加SENet代码,代码如下：

#SENet 自己添加的
class SE(nn.Module):
    def __init__(self, c1, c2, ratio=16):
        super(SE, self).__init__()
        #c*1*1
        self.avgpool = nn.AdaptiveAvgPool2d(1)
        self.l1 = nn.Linear(c1, c1 // ratio, bias=False)
        self.relu = nn.ReLU(inplace=True)
        self.l2 = nn.Linear(c1 // ratio, c1, bias=False)
        self.sig = nn.Sigmoid()
    def forward(self, x):
        b, c, _, _ = x.size()
        y = self.avgpool(x).view(b, c)
        y = self.l1(y)
        y = self.relu(y)
        y = self.l2(y)
        y = self.sig(y)
        y = y.view(b, c, 1, 1)
        return x * y.expand_as(x)

2 仍然在此文件下添加如下内容。