yolov5数据集_140FPS!YOLOv5重磅来袭,速度逆天

28c591c8f58d32fd26bc044c506decc4.png

在YOLOv4发布后不到50天,YOLOv5就改进了实时对象检测的技术水平。

实时目标检测技术快速发展,其改进的速度令人震惊。

3月18日,Google开源了EfficientDet的实现源代码,EfficientDet可对不同大小的模型进行快速训练,其中之一提供实时输出。4月23日,Alexey Bochoviskiy等人开源了YOLOv4。6月9日,Glenn Jocher开源了 YOLOv5。

6ef6b894ae5728738ce1e208e85682f6.gif

YOLO模型的演变

YOLO(You Only Look Once)是PJ·雷迪(PJ Reddie)最初在2016年出版的一本书中创建的一系列模型。YOLO模型非常小,以高性能著称,这使它们成为实时条件和设备上部署环境的理想候选者。

PJ Reddie的研究团队随后开源了YOLOv2和YOLOv3,这两个产品在模型性能和模型速度上都有了持续的提升。

2020年2月,PJ Reddie表示他将停止计算机视觉的研究。2020年4月,Alexey Bochkovskiy, Chien-Yao Wang和hongyuan Mark Liao推出了YOLOv4,取得了令人瞩目的成绩。

1d6a44dfa38926708ca6e779a1085dec.png

值得注意的是,YOLOv4的许多改进来自于改进的数据增强和模型架构。

YOLOv5:实时目标检测的领导者

Glenn Jocher发布的YOLOv5与此前的YOLO模型相比存在很多差异和改进。

值得注意的是,Glenn是mosaic 数据增强方法的创建者,该方法是改进后的YOLOv4中包含的一种技术。YOLOv5的发布包括五个不同的型号:YOLOv5s(最小)、YOLOv5m、YOLOv5l、YOLOv5x(最大)。

首先,这是YOLO系列中第一个使用PyTorch而不是PJ Reddie的Darknet编写的native版本。

Darknet是一个非常灵活的研究框架,但它并不是在考虑生产环境的情况下构建的,它的用户群体也比较小。因此Darknet在配置上更具有挑战性,且不适合生产。

因为YOLOv5是在PyTorch中实现的,所以它受益于已建立的PyTorch生态系统;支持更简单,部署更容易。

此外,作为一个更广为人知的研究框架,在YOLOv5上进行迭代可能更容易。

由于该模型可以轻松地编译为ONNX和CoreML,因此这也使得部署到移动设备的过程更加简单。

其次,YOLOv5极快。在运行Tesla P100的YOLOv5 Colab笔记本中,每个图像的推理时间快至0.007秒,意味着每秒140帧(FPS)!相比之下,YOLOv4在转换为相同的Ultralytics PyTorch后达到了50帧 。

第三,YOLOv5是准确的。在研究者对血细胞计数和检测(BCCD)数据集的测试中,经过100个epoch的训练,他们获得了大约0.895的平均精度(mAP)。研究者也在EfficientDet和YOLOv4上也看到了相当的性能,但是很少看到YOLOv5这种全面的性能改进而又不损失准确性。

第四,YOLOv5很小。具体而言,YOLOv5的权重文件为27兆字节。YOLOv4(具有Darknet架构)的权重文件为244 MB。YOLOv5比YOLOv4小近90%。这意味着YOLOv5可以更轻松地部署到嵌入式设备。

15d3581eab638e7b332d56012e683231.png
YOLOv5权重文件存储在S3中,以便将来进行推断

YOLOv5的图形测量性能很好地总结了这些变化。

ac28109ea24393345a4bb4842b4242c6.png
YOLO比EfficientDet更快、更准

参考链接:https://blog.roboflow.ai/yolov5-is-here/

github:https://github.com/ultralytics/yolov5

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值