目录
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