1、前言
yolov5目录中,关于模型构建的为下面的部分:
- *.yaml 文件,具体的配置参数
- common文件,具体模块的搭建,neck的spp等等
- yolo文件,搭建yolo的代码
这部分为了后续添加改进模块,所以详细介绍下
2、yaml 文件
因为之前实战的模型都是v5s,这里我们打开yolov5s.yaml文件查看,如下:
# YOLOv5 🚀 by Ultralytics, AGPL-3.0 license
# Parameters
nc: 80 # number of classes
depth_multiple: 0.33 # model depth multiple
width_multiple: 0.50 # layer channel multiple
anchors:
- [10, 13, 16, 30, 33, 23] # P3/8
- [30, 61, 62, 45, 59, 119] # P4/16
- [116, 90, 156, 198, 373, 326] # P5/32
# YOLOv5 v6.0 backbone
backbone:
# [from, number, module, args]
[
[-1, 1, Conv, [64, 6, 2, 2]], # 0-P1/2
[-1, 1, Conv, [128, 3, 2]], # 1-P2/4
[-1, 3, C3, [128]],
[-1, 1, Conv, [256, 3, 2]], # 3-P