点击上方“AIWalker”,选择加“星标”或“置顶”
重磅干货,第一时间送达
❝今天在刷github时,突然看到了YOLOv5,笔者当时还在怀疑是不是眼花了?确实是YOLOv5,但不是官方的也不是AB大神版,而是U版YOLO改进版。哎,想想真可怜,笔者还在熟悉YOLOv4的时候,YOLOv5竟然出现了,太快了,跟不上节奏啊。不过不妨碍我们去研究了解它。
❞
关于前段时间刷屏的YOLOv4,笔者之前有过介绍,见链接:https://zhuanlan.zhihu.com/p/135840896 感兴趣的小伙伴可以去看看。
严格来讲,这里提到的YOLOv5可以视为YOLOv4.5,它仍然是各种tricks的组合,但更关键的是它的实现方式由darknet转为了pytorch,同时还支持ONNX以及CoreML。真乃天大的福音啊,还在等什么么?还不快点去围观关注一波!
YOLOv5代码链接:https://github.com/ultralytics/yolov5 实现者是谁呢?Ultralytics:https://www.ultralytics.com,大家也许听过U版yolo,即:U版YOLOv3 https://github.com/ultralytics/yolov3,见下图拥有4.7k-star的U版YOLOv3,所以你就知道了吧,吼吼
image-20200530154457401言归正传,我们先来看看YOLOv5的一些“官方”说明吧。首先给出了已开源的YOLOv5在COCO2017数据集上的指标,见下图。可以看到,它基本上超过了大部分的目标检测器,同时具有超快的速度。
image-20200530152827837注1:指的是COCO-test-dev2017上的结果,其他AP表示val2017的精度。所有指标均为单模型、单尺度、为进行任何集成与TTA。可以通过如下命令进行重现上述指标,非常赞啊!
python test.py --img-size 736 --conf_thres 0.001
注2:上述延迟是在V100 GPU上对5000张COCO-val2017数据进行测试的结果,包含数据预处理、模型推理、后处理以及NMS。平均NMS的时间为1.6ms/image。上述指标可以通过如下命令复现:
python test.py --img-size 640 --conf_thres 0.1 --batch-size 16
注3:上述所有模型均采用默认配置与超参数训练了300epoch,且未添加自动增广技术。
下面附上训练曲线与测试效果。
image-20200530153650522 image-20200530153601144附上一些Ultralytics的信息,摘自:https://github.com/ultralytics/yolov5,为避免凸显笔者粗陋的英语水平,就直接上英文原文了,哈哈哈
Ultralytics is a U.S.-based particle physics and AI startup with over 6 years of expertise supporting government, academic and business clients. We offer a wide range of vision AI services, spanning from simple expert advice up to delivery of fully customized, end-to-end production solutions, including:
最后附上YOLOv5项目的一些更新说明,同上,附上英文原本,哈哈。
This repository represents Ultralytics open-source research into future object detection methods, and incorporates our lessons learned and best practices evolved over training thousands of models on custom client datasets with our previous YOLO repository https://github.com/ultralytics/yolov3. 「All code and models are under active development, and are subject to modification or deletion without notice.」 Use at your own risk.
Updates:
虽然笔者不是专职目标检测方向的工作,但对YOLO还真听过非常多,笔者带过的团队也用过YOLOv3方法进行目标检测相关应用;虽然YOLOv5仍处于研发阶段,尚未如YOLOv4得到官方的认可,但YOLOv5的指标仍然非常棒,也值得各位CVer去研究与应用,关键它是pytorch。最后期待U版YOLO继续开展后续的研究、贡献更多优质的模型,感兴趣的CVer还不赶快去“素质三连”一波?吼吼!
当然,最重要的是:关注一下AIWalker并添加星标-置顶,关注一下AIWalker并添加星标-置顶,关注一下AIWalker并添加星标-置顶。重要的事情说三遍!
欢迎关注AIWalker公众号,在这里您将得到独家深度学习经验分享与个人思考。想支持Happy继续写下去就点个赞关注一下吧!