YOLOv5配置了4种不同大小的网络模型,分别是YOLOv5s
、YOLOv5m
、YOLOv5l
、YOLOv5x
,其中 YOLOv5s
是网络深度和宽度最小但检测速度最快的模型,其他3种模型都是在YOLOv5s
的基础上不断加深、加宽网络使得网络规模扩大,在增强模型检测性能的同时增加了计算资源和速度消耗。出于对检测精度、模型大小、检测速度的综合考量,本文选择YOLOv5s
作为研究对象进行介绍。
./models/yolov5s.yaml
文件是YOLOv5s网络结构的定义文件,如果你想改进算法的网络结构,需要先修改该文件中的相关参数,然后再修改./models/common.py
与./models/yolo.py
中的相关代码。
- 📌 本周任务:将yolov5s网络模型中第4层的
**C3*2**
修改为**C3*1**
,第6层的**C3*3**
修改为**C3*2**
。 - 💫 任务提示:仅需修改
./models/yolov5s.yaml
文件。