0.引言
由于YOLOv5/YOLOv7使用的设备不尽相同，考虑控制变量法，特此写一篇博客记录一下各模型的横向对比（由于时间有限，因此只针对640尺寸的模型进行训练测试）
1.软硬件配置
（1）硬件配置
- CPU：英特尔E5-2640 v3@2.6GHz
- GPU：Tesla P40 24G*4
- 内存：64G
（2）软件配置
- Ubuntu18.04
- cuda11.3
- pytorch:1.11.0
- torchvision:0.12.0
2.数据集配置
2020年kaggle小麦检测数据集，包含3187张图片，各类标签的数量分别为：
- arvalis_1: 42962
- arvalis_2: 4022
- arvalis_3: 15715
- rres_1: 19077
- ethz_1: 48606
- usask_1: 5605
- inrae_1: 3540
3.不同模型性能对比表
每种算法均按默认配置，使用小麦检测数据集训练100轮，batch_size按显存最大来设置，img_size均为640*640，使用train.py进行4卡训练，使用detect.py进行单卡预测。加粗的表示最优结果。
| 算法 | batch_size | param/Million | FLOPs/G | weight_size/MB | P/% | R/% | mAP50/% | mAP50-95/% | train_time/h | Speed/ms |
|---|---|---|---|---|---|---|---|---|---|---|
| yolov5n | 256 | 1.7 | 4.3 | 3.9 | 91.9 | 88.1 | 93.9 | 53.2 | 0.682 | 11.0 |
| yolov5s | 256 | 7.0 | 16.0 | 14.5 | 92.7 | 90.3 | 94.8 | 55.6 | 0.705 | 13.0 |
| yolov5m | 128 | 20.9 | 48.3 | 42.3 | 93.1 | 89.4 | 94.2 | 55.0 | 1.0098 | 16.8 |
| yolov5l | 64 | 46.2 | 108.3 | 92.9 | 93.1 | 88.8 | 94.3 | 55.0 | 1.751 | 25.6 |
| yolov5x | 32 | 86.2 | 204.8 | 173.2 | 92.6 | 89.4 | 94.5 | 55.4 | 3.068 | 40.4 |
| – | – | – | – | – | – | – | – | – | – | – |
| yolov7-tiny | 512 | 6.03 | 13.2 | 12.3 | 64.4 | 61.2 | 68.8 | 29.1 | 1.130 | 11.5 |
| yolov7 | 128 | 37.2 | 105.2 | 74.9 | 92.0 | 91.9 | 94.9 | 54.9 | 1.912 | 30.4 |
| yolov7x | 128 | 70.9 | 189.0 | 142.2 | 93.4 | 91.0 | 94.6 | 54.9 | 2.668 | 42.3 |
| – | – | – | – | – | – | – | – | – | – | – |
| yolov8n | 256 | 3.1 | 8.2 | 6.3 | 91.9 | 88.4 | 94.1 | 55.6 | 1.641 | 14.7 |
| yolov8s | 128 | 11.1 | 28.7 | 22.5 | 91.1 | 89.0 | 94.2 | 56.1 | 1.862 | 13.9 |
| yolov8m | 128 | 25.9 | 79.1 | 52.1 | 91.6 | 90.0 | 94.6 | 56.4 | 2.228 | 20.3 |
| yolov8l | 64 | 43.6 | 165.4 | 87.7 | 92.1 | 89.1 | 94.6 | 56.6 | 2.974 | 30.7 |
| yolov8x | 64 | 68.2 | 258.2 | 136.8 | 91.8 | 90.3 | 95.0 | 56.8 | 3.658 | 40.2 |
4.结论
- 从同一模型来看，模型的准确率和召回率并没有随着模型的加深而提高，相反，有着一定的降低，这主要是由于数据集量相比COCO数据集小得多，模型中有着大量的冗余参数，导致模型的各项指标下降；因此，对于量较小的数据集，选用参数量较少的模型，不光可以提高准确率，降低训练时间，还能在预测速度上有着比较大的优势；
- 对比不同模型来看，yolov8相较yolov5和yolov7在准确率方面确实有一定的提升（仅限于大模型l/x），在本文使用的数据集上提升大概在1个点左右；而小模型（n/s）其实差距不大，甚至yolov7-tiny远低于正常水平（不知道是否是训练的时候出问题了，后续需要再测一下），这是我没有想到的；
- 需要根据实际情况挑选自己的模型，没有最好的模型，只有最适合自己数据集的模型。
5.后记
- 如果还不会使用yolov8进行训练的朋友可以移步这篇文章：
YOLOv8系列教程：一、使用自定义数据集训练YOLOv8模型（详细版教程），包含环境搭建/数据准备/模型训练/预测/验证/导出等 - 如果想使用yolov7进行训练的朋友可以看下这几篇文章：
YOLOv7系列教程：一、基于自定义数据集训练专属于自己的目标检测模型（保姆级教程，含数据集预处理），包含对train.py/test.py/detect.py/export.py详细说明
YOLOv7系列教程：二、使用onnx导出成tensorrt模型，实现高性能部署，包含opencv多线程、图像队列存取、自动保存xml和jpg等模块
YOLOv7系列教程：三、使用onnx导出成openvino模型，并调用接口进行摄像头推理预测
文章出处登录后可见！
已经登录？立即刷新
