YOLOv4还没有退热,YOLOv5已经发布!
6月9日,Ultralytics公司开源了YOLOv5,离上一次YOLOv4发布不到50天。而且这一次的YOLOv5是完全基于PyTorch实现的!
在我们还对YOLOv4的各种骚操作、丰富的实验对比惊叹不已时,YOLOv5又带来了更强实时目标检测技术。
按照官方给出的数目,现版本的YOLOv5每个图像的推理时间最快0.007秒,即每秒140帧(FPS),但YOLOv5的权重文件大小只有YOLOv4的1/9。
接下来一起了解一下最新的YOLOv5吧。
更快更小巧的YOLOv5
老规矩,先来介绍一下安装过程。
安装YOLOv5通过下方代码实现:
$ pip install -U -r requirements.txt
要求Python3.7以上版本,PyTorch1.5以上。
YOLOv5的4组不同权重在COCO2017数据集上的结果如下:
所有模型均采用默认配置与超参数训练了300epoch,且未添加自动扩增技术。
可以看到,它超过了大部分的目标检测器,同时具有超快的速度。
APtest指的是COCO-test-dev2017上的结果,其他AP表示val2017的精度。所有指标均为单模型、单尺度、为进行任何集成与TTA。可以通过如下命令进行重现上述指标:
python test.