看完这篇你能学会什么?
- 掌握根据
yaml
文件画出模型结构图的能力 - 掌握根据模型结构图写
yaml
文件的能力 - 掌握添加模块后写配置文件
args
参数的能力 - 掌握修改模型配置文件的能力
1. YOLOv5 模型yaml文件解析
模型 | 尺寸 (像素) | mAPval 50-95 | mAPval 50 | 推理速度 CPU b1 (ms) | 推理速度 V100 b1 (ms) | 速度 V100 b32 (ms) | 参数量 (M) | FLOPs @640 (B) |
---|---|---|---|---|---|---|---|---|
YOLOv5n | 640 | 28.0</ |