YOLOV5目标检测----上手

YOLOV5目标检测----上手

在这里插入图片描述

2020年4月23日,YOLOV4开源

2020年6月10日,YOLOV5开源

这两个日子,可是把CV类公众号的运营者忙坏了,各大公众号都在忙着抢“首发”

当然,YOLOV4是货真价实的YOLO系列的第四个版本,虽然J.R大神宣布了“退役”,但是v4依然是其团队的研究成果,是具有正宗YOLO血统的

对于YOLOV5,其作者显然不是YOLO团队的,但是v5的作者也是yolo系列忠实粉丝。ultralytics团队实现的pytorch版本的yoloV3,广受好评,在github上获得了5.2K的star。

在YOLOV5开源这一天,其repo上就有人发了issue,说不应该使用yoloV5这个名字。
在这里插入图片描述

在下面讨论区,两派的声音势均力敌。
在这里插入图片描述

这个“you can you up“,简直笑到我了!
在这里插入图片描述

对于我来说,这个命名也就是一个代号,不管你叫啥,只要性能牛逼,我就给你点star


一、撕逼

我们现在回到YOLOV5这个算法上来
在这里插入图片描述
上图来自于yoloV5的repo,给出了不同大小v5模型和efficientDet的比较。

从图中可以看出,yoloV5不同大小的模型,其性能都优于efficientDet,但是有一点大部分人都觉得很奇怪,为啥没有跟yoloV4比较。既然名字都命名了V5,为啥没有与V4的对比实验。

YoloV5 is here这篇博客中说了YOLOv5在Tesla P100显卡上运行超过140FPS,YOLOv4只能运行到50FPS;v5的模型大小是v4模型大小的十分之一,并且两者精度相当。

感兴趣的同学可以看下原文,截图如下。
在这里插入图片描述

到目前为止,YOLOV5还没有发表学术论文,ultralytics是一个公司,并不是一个学术组织。所以现在有很多人对github上发出来的对比数据是持怀疑态度的。详情可以看讨论区社区

甚至,AlexeyAB也直接在issue怼上了
在这里插入图片描述

AlexeyAB的态度很明确,u团队做对比实验不科学。拿着小模型的v5和大模型的v4比速度,然后也没提供输入分辨率和batch_size。然后AlexeyAB自己做了对比实验,数据如上图。

然后 Ultralytics团队的Glenn- Jocher,对此作出了回应。表示他们原有计划是要写一篇paper来展示实验过程的,但是由于公司资源限制,这个计划被推迟了。毕竟公司是要恰饭的,可以理解。
在这里插入图片描述

关于YoloV4和YoloV5的对比测评,可以看RoboFlow写的这篇文章。这篇文章从yolo的历史开始写起,讨论了命名和自定义数据下的性能。

下图是AlexeyAB实验对比数据,分别对比了v3,v4和v5
在这里插入图片描述

上面的内容基本都是围绕着命名和性能在讨论(撕逼)

但是对于我们算法狗而言,无所谓啦,只要能开源,你叫啥都行!

把开源的变成自己的,把自己的变成祖传的。

二、训练自己的yoloV5

在这里插入图片描述

1、配置环境

git clone https://github.com/ultralytics/yolov5
pip install -U -r yolov5/requirements.txt

2、转换数据格式

将自己的xml或者json文件转换为yolo格式的txt文件

3、选择模型大小

可以选择yolo5s,yolo5m,yolo5l,yolo5x不同大小的模型,模型越大精度越高,速度也就相应越慢

4、修改对应配置文件

  • data.yaml
  • custom_yolov5m.yaml

训练过程可以参考yoloV5的wiki,也可以参考这个训练repo另外一个训练教程

v5的训练速度还是很快的

我原来有个目标检测项目是使用pelee做的,由于硬件算力限制,不得已只能选轻量级的算法。所以yoloV5s和yolov5m对于轻量级算法选型应该是一个不错的选择。

在这里插入图片描述

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章