1、前言
本章将介绍yolov5的改进项目,为v5增加新的模块---注意力机制、SE模块
大部分更改的代码是重复的,只有少部分需要更改,下面会详细讲解
yolov5的yaml文件介绍:YOLOV5 模型:利用tensorboard查看网络结构和yaml文件介绍-CSDN博客
yolov5的模块更改,C3更改为C2f模块:
YOLOV5 改进:修改网络结构--C2f 模块-CSDN博客
2、SE 模块介绍
SE模块的介绍:主要包含Squeeze和Excitation两部分
主要的思想就是为不同的像素点增加权重,因为人眼在观察物体进行识别的时候,对不同部分侧重点是不同的,所以注意力机制就是为了模拟这一特点
Tips : 这里的输入和输出维度是一样的!