文章目录[隐藏]
注:个人水平有限,如有错误(以下内容加入自己的理解),敬请批评指正。
YOLOX论文
论文:https://arxiv.org/pdf/2107.08430.pdf
github地址:https://github.com/Megvii-BaseDetection/YOLOX
摘要
YOLOX在YOLO系列中加入了一些技巧,如
1.将YOLO检测器和anchor-free结合
2.加入了decoupled head
3.加入了SimOTA
和其他方法的比较:
- YOLO-Nano参数仅有0.91M和1.08GFLOPS,在COCO数据集上有25.3%AP,相比NanoDet,提高了1.8%
- 在COCO上将工业上常用的YOLOv3提高到了47.3%
- YOLOX-L的参数量和YOLOv5-L的参数量大致相同,YOLOX-L在V100(我:V100是多少显存的?估计应该是32GB的吧)上速度为68.9FPS,AP超过YOLOv5-L大约1.8%
在CVPR2021的Streaming Perception Challenge中,使用YOLOX-L模型获得第一名。
为了让模型更好落地,论文提供了ONNX,TensorRT,NCNN和Openvino的部署版本的github地址
(我:后面我会试试部署版本)
1.介绍
虽然YOLO系列在精度和速度方面都蛮好的。
但是,最近几年目标检测领域的“骚操作”,如anchor-free检测器、更优秀的label assignment技巧、端到端检测器(NMS-free)都没有被集成到YOLO系列中。这篇文章就是做这个的,YOLOv4和YOLOv5太优秀了(手动狗头.jpg),先从YOLOv3“虐”起吧。看看下面这幅图吧,一图胜千言:
2.YOLOX
2.1 YOLOX-DarkNet53
训练细节
去论文里看吧,略过
YOLOv3 baseline
baseline使用darknet53作为主干特征提取网络,并加入了SPP层。
相比原始的YOLOv3,本论文的YOLOv3 baseline对训练策略做了如下改动:
1.加入EMA权重更新
2.加入cosine lr schedule
3.加入IoU loss
4.加入IoU-aware branch
YOLOv3 baseline结果如下:
Decoupled head
YOLO系列的主干特征网络和特征金字塔不断进化,他们的detection head耦合在一起,detection head耦合在一起会降低性能。
对性能的影响程度?给你两张图片,自己品,你细品
数据增强
使用Mosaic和MixUp增强图像数据集,借此提高YOLOX性能。
使用数据增强后,抛弃了ImageNet的预训练模型,从头开始训练模型。
Achor-free
YOLOv4和YOLOv5都是anchor-based,但是anchor-based有如下缺点:
1.训练之前,需要进行聚类分析决定最优的anchors,这些anchors泛化性能弱
2.anchors增加了检测的复杂性
Multi positives
positives的处理类似FCOS
SimOTA
SimOTA应该是Simplified Optimal Transport Assignment
优秀的label assignment应该具备的特征:
1.loss/quality aware
2.center prior
3.dynamic top-k
4.global view
OTA满足上面4个条件,但是考虑到训练时间问题,使用SimOTA
端到端的YOLO
实现方法:
1.加入两层额外卷积层
2.加入一一映射的label assignment
3.加入stop gradient
2.2 其他backbone
-
YOLOX v.s. YOLOv5
-
YOLOX-Tiny&Nano v.s. other-Tiny&Nano
-
不同模型大小适用不同的数据增强策略
3.与其他方法的比较
“是骡子是马,拉出来溜溜”
下表是不同模型在COCO2017 test-dev上的比较,公平起见,都是300 epoch
4.SPC比赛中第一名
Streaming Perception Challenge第一名,有兴趣的话去看看。
5.总结
本文提出了YOLOX,它集成了如下技巧改进YOLO系列:
1.decoupled head
2.anchor-free
3.优秀的label assigning策略
文章下期预告
YOLOX的部署
文章下下期预告
YOLOX的自定义部署
不说了,我先打麻将去了(模仿FJYB)
版权声明:本文为CSDN博主「机器视觉全栈er」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/qq_42322191/article/details/118936680
暂无评论