YOLOv5来了!基于PyTorch,体积比YOLOv4小巧90%,速度却超2倍

新闻 前端
YOLOv5来了!基于PyTorch,体积只有YOLOv4的十分之一,速度近3倍,权重可以导出到移动端,并且在COCO上达到了最先进的水平。

 来了,来了,YOLOv5来了!

Ultralytics正式更新了YOLOv5,已经登顶GitHub飙升榜首席。

       

它运行推理的速度极快,权重可以导出到移动端,并且在COCO上达到了最先进的水平。

       

 YOLOv5项目由Ultralytics创建并维护。这是一家Ultralytics总部位于美国的粒子物理和人工智能初创公司,为政府、学术和商业客户提供支持。在粒子物理学、数据科学和人工智能领域率先提出了多项美国情报界(IC)和国防部(DoD)的倡议,并与多所大学、国家实验室和私营公司合作开展了多项工作。      

 

你肯定也注意到了iOS这个关键词。他们确实做了一个iOS app,还特意做了一张海报,简直让人印象深刻,大家来感受一下:

 要不是左下角的「App Store认证标志」,我还以为是某手机的广告呢。毕竟现在的手机都长得像iPhone,电脑都长得像Mac。

YOLO模型的演变

 实时目标检测的改进速度很快。改进的速度更是迅速。结果也是惊人的。

 3月18日,谷歌开源了EfficientDet的实现,这是一个具有不同大小的快速训练模型,其中一个模型提供实时输出。

 4月23日,Alexey Bochoviskiy等人开源了YOLOv4。

 6月9日,Glenn Jocher开源了YOLOv5的PyTorch实现。

 YOLO(You Only Look Once)是PJ Reddie最初以2016年的一篇文章创造的一个模型家族。YOLO一出生便技惊四座。人们简直不敢相信小小体积的YOLO竟然能表现出如此超高的性能!简直就是实时条件和设备上部署环境的理想候选人。

 PJ Reddie的研究团队负责随后推出了YOLOv2和YOLOv3,这两个模型在性能和速度上都做出了持续改进。2020年2月,PJ Reddie指出他将停止计算机视觉的研究。

 2020年4月,Alexey Bochkovskiy、Chien-Yao Wang和Hong-Yuan Mark Liao推出了YOLOv4,展示了令人印象深刻的成果。

值得注意的是,YOLOv4的许多改进来自于改进的数据增强和模型架构。我们已经写了关于YOLOv4的分析,以及如何在自定义对象上训练YOLOv4模型)。

YOLOv5:实时目标检测的新时代领航者

相比其他版本,Glenn Jocher发布的YOLOv5有许多不同之处和改进。Glenn是YOLOv4中一个非常关键的技术「mosaic augmentation」的创建者。

YOLOv5并不是一个单独的模型,而是一个模型家族,包括了YOLOv5s(最小)、YOLOv5m、YOLOv5l、YOLOv5x(最大)。

那么,YOLO5凭什么能够成为目标检测新的领航者呢?RoboFlow介绍了4点优势。

首先,这是YOLO家族中,第一个先用PyTorch的原生版本,而非PJ Reddie的Darknet编写的模型。Darknet是一个非常灵活的研究框架,但它并没有考虑到生产环境的构建,而且用户社区也较小。这导致Darknet需要在配置上花费不少功夫,而且生产准备不足。

由于YOLOv5是在PyTorch中实现的,它受益于成熟的PyTorch生态系统:支持更简单,部署更容易。此外,作为一个更广为人知的研究框架,YOLOv5 的迭代对更广泛的研究社区来说可能更容易。这也使得部署到移动设备上更加简单,因为该模型可以轻松编译成ONNX和CoreML。

其次,YOLOv5的速度快得惊人。在YOLOv5 Colab notebook上,运行Tesla P100,我们看到每张图像的推理时间仅需0.007秒,这意味着每秒140帧(FPS)!相比之下,YOLOv4在被转换到相同的Ultralytics PyTorch库后的速度是50FPS。YOLOv5的速度是YOLOv4的2倍还多!

第三,YOLOv5精度超高。在Roboflow对血细胞计数和检测(BCCD)数据集的测试中,只训练了100个epochs就达到了大约0.895的平均精度(mAP)。诚然EfficientDet和YOLOv4的性能相当,但在准确率没有任何损失的情况下,看到如此全面的性能提升是非常罕见的。

第四,YOLOv5的体积很小。具体来说,YOLOv5的权重文件是27兆字节。YOLOv4(采用Darknet架构)的权重文件是244兆。YOLOv5比YOLOv4小了近90%!这意味着YOLOv5可以更容易地部署q到嵌入式设备上,所以Ultralytics也在招iOS开发。

项目地址:

https://github.com/ultralytics/yolov5

 

责任编辑:张燕妮 来源: 新智元
相关推荐

2020-10-05 21:57:17

GitHub 开源开发

2024-10-07 11:12:55

2024-08-06 11:01:03

2024-06-20 09:56:28

2020-04-30 20:54:11

Windows漏洞网络安全

2023-07-18 14:50:15

2020-10-26 14:14:36

YOLOv5数据科学 技术

2023-09-29 11:55:55

2021-12-08 09:47:21

安卓系统模型

2023-01-10 18:33:18

AIYOLO网络

2020-06-11 10:05:00

计算机视觉 技术

2022-04-26 15:24:03

开源框架

2023-03-16 19:17:57

2015-09-10 11:29:57

5G4G

2020-10-19 10:40:45

5G网络技术

2018-01-29 05:38:20

5G4G运营商

2022-12-13 14:46:25

模型

2023-01-03 08:42:16

2020-06-28 10:16:53

PyTorchTensorFlow机器学习

2009-07-01 09:47:10

FireFox 3.5浏览
点赞
收藏

51CTO技术栈公众号