YOLOV5 改进：替换backbone为GhostNet

目录

1、介绍

2、更改common文件 

3、更改yolo脚本

4、创建yaml文件

5、测试一下是否添加成功

6、训练

---

1、介绍

YOLOV5 替换GhostNet 主干网络的顺序如下：

2、更改common文件 

将下面代码加入common最下面即可：

import torch
import torch.nn as nn


# Squeeze-and-Excitation (SE) 模块定义
class SeBlock(nn.Module):
    def __init__(self, in_channel, reduction=4):
        super().__init__()

        # Squeeze：全局平均池化层
        self.Squeeze = nn.AdaptiveAvgPool2d(1)

        # Excitation：两层全连接层（1x1卷积）
        self.Ex