
以下是这些 YOLO 系列配置文件的详细解析,按版本和功能分类说明:
一、YOLOv3 系列
文件名 |
核心特性 |
适用场景 |
yolov3.yaml |
原始 YOLOv3 结构,3 尺度预测(13x13,26x26,52x52) |
通用目标检测 |
yolov3-spp.yaml |
增加 SPP(空间金字塔池化)模块,提升感受野 |
大目标检测(如遥感图像) |
yolov3-tiny.yaml |
轻量化版本,仅保留 2 个预测层,通道数减半 |
边缘设备实时检测 |
二、YOLOv5 基础系列
按模型大小分类(参数量递增):
文件名 |
模型规模 |
参数量(M) |
FLOPs(G) |
特点 |
yolov5n6.yaml |
Nano |
1.9 |
4.5 |
最小模型,适合移动端 |
yolov5s6.yaml |
Small |
7.2 |
16.5 |
平衡速度与精度 |
yolov5m6.yaml |
Medium |
21.2 |
49.0 |
主流选择 |
yolov5l6.yaml |
Large |
46.5 |
109.1 |
高精度场景 |
yolov5x6.yaml |
XLarge |
86.7 |
205.7 |
最大模型,需高性能GPU |
注:后缀 6
表示输入分辨率为 640x640(默认 yolov5s
为 640x640,yolov5s6
强调分辨率配置)
三、YOLOv5 结构变体
1. 特征金字塔改进
文件名 |
结构特性 |
yolov5-fpn.yaml |
使用标准 FPN(自顶向下融合) |
yolov5-panet.yaml |
使用 PANet(双向特征金字塔) |
yolov5-bifpn.yaml |
采用 BiFPN(加权双向跨尺度连接) |
2. 多尺度预测增强
文件名 |
预测层配置 |
优势领域 |
yolov5-p2.yaml |
2 层预测(P3, P4) |
中大型目标快速检测 |
yolov5-p34.yaml |
3 层预测(P3, P4, P5) |
通用场景 |
yolov5-p6.yaml |
4 层预测(P3-P6),最高层下采样64倍 |
极小目标检测(如无人机) |
yolov5-p7.yaml |
5 层预测(P3-P7),支持 1280x1280 输入 |
超高清图像分析 |
3. 模块替换优化
文件名 |
关键技术 |
作用 |
yolov5s-ghost.yaml |
使用 Ghost 模块替代常规卷积 |
减少 30% 计算量 |
yolov5s-LeakyReLU.yaml |
激活函数替换为 LeakyReLU(原版为 SiLU) |
兼容老旧硬件部署 |
yolov5s-transformer.yaml |
在 Neck 部分引入 Transformer 编码器 |
提升复杂场景识别能力 |
四、锚点框配置
文件名 |
作用 |
anchors.yaml |
定义锚点框(anchor boxes)的尺寸和比例,需根据数据集目标尺寸调整 |
五、典型场景推荐
需求 |
推荐配置文件 |
硬件要求 |
工业质检(小缺陷) |
yolov5m6-p6.yaml |
RTX 3080 |
移动端实时检测 |
yolov5n6.yaml |
Jetson Nano |
自动驾驶(多尺度目标) |
yolov5x6-bifpn.yaml |
A100 GPU |
兼容 TensorRT 部署 |
yolov5s-LeakyReLU.yaml |
支持 FP16 的嵌入式设备 |
关键参数对比示例
以输入尺寸 640x640 为例:
模型 |
mAP@0.5 |
速度(FPS) |
显存占用(GB) |
yolov5n6 |
28.9 |
220 |
1.2 |
yolov5s-bifpn |
37.5 |
120 |
3.8 |
yolov5x6-p7 |
52.1 |
35 |
18.4 |
注:数据基于 COCO 2017 验证集,测试硬件为 RTX 3090。