美团出品 | YOLOv6 v3.0 is Coming(超越YOLOv7、v8)

美团出品 | YOLOv6 v3.0 is Coming(超越YOLOv7、v8)

🚀🚀🚀美团出品 | YOLOv6 v3.0 is Coming !!✨✨✨

一、前言简介 🎄🎈

📚 代码地址:美团出品 | YOLOv6 3.0代码下载地址

📚 文章地址:https://arxiv.org/pdf/2301.05586.pdf

美团出品 | YOLOv6 v3.0 is Coming(超越YOLOv7、v8)

🎄🎄YOLOv8才刚刚发布!美团出品 | YOLOv6 v3.0它来啦!其优越性能已超越当下主流检测器YOLOv7、v8!!!🎈🎈

美团出品 | YOLOv6 v3.0 is Coming(超越YOLOv7、v8)

YOLOv6 3.0亮点贡献:

1.更好的网络结构设计:使用双向串联Bi-directional Concatenation (BiC)模块更新了检测器的颈部,以提供更准确的定位信号。同时,将 YOLOv5/v8 中的SPPF模块简化为 SimCSPSPPF模块,在速度几乎保持不变的同时提升精度。

2.辅助训练分支:提出了一种 Anchor-Aided Training的辅助训练(AAT)策略,以在不影响推理效率的情况下同时享受到 Anchor-Based 和 Anchor-Free范式的优势。

3.网络层数加深:在更高分辨率输入下达成新的SOTA性能。

4.更好的蒸馏策略:在回归分支中使用DFL(Distribution Focal Loss)做蒸馏学习;由于DFL的引入会导致reg分支结构变大,因此在小模型中DFL只用于辅助训练,推理时不使用DFL,以提升小模型的推理速度。

🍀相关实验效果:

美团出品 | YOLOv6 v3.0 is Coming(超越YOLOv7、v8)

二、创新改进模块 🎄🎈

(一)BiC模块以及SimCSPSPPF模块

作者在YOLOv6 3.0使用了增强型的PAN作为Neck,网络总体结构如下图所示。同时,使用了Bi-directional Concatenation(BiC)来融合Backbone输出的相邻层的特征。BiC还使用了backbone中的特征,以保留更精确的定位信息,利于小目标的检测。并将SPPF block与CSP进行结合,并简化了SPPF block,如下图图(c)所示。

美团出品 | YOLOv6 v3.0 is Coming(超越YOLOv7、v8)

(二)Anchor-Aided Training

YOLOv6是一种追求更高推理速度的Anchor-free检测器,但作者在实验中发现:在同等配置(YOLOv6-N)下,相比Anchor-free方案,Anchor-based方案可以带来额外的性能增益。

美团出品 | YOLOv6 v3.0 is Coming(超越YOLOv7、v8)

作者提出了一种 Anchor-Aided Training的辅助训练(AAT)策略,引入了一个Anchor-based辅助分支以组合两种方案的优势。通过这种训练策略,来自辅助分支的引导信息可以被有效的嵌入到Anchor-free分支。在推理阶段,辅助分支将被移除掉。

美团出品 | YOLOv6 v3.0 is Coming(超越YOLOv7、v8)

(三)自蒸馏策略

作者在分类蒸馏的基础上,还引入了DFL(Distribution Focal Loss)用于回归分支的蒸馏。

美团出品 | YOLOv6 v3.0 is Coming(超越YOLOv7、v8)

美团出品 | YOLOv6 v3.0 is Coming(超越YOLOv7、v8)

作者采用了这种新的自蒸馏策略DFL,提高了 YOLOv6小模型的性能,其中 DFL 的较重分支在训练期间被用作增强的辅助回归分支,并在推理时被移除以避免推理速度下降。

相关参考链接:

Beyond YOLOv8!| YOLOv6 v3.0 is coming!

YOLOv6 v3.0 | 重磅升级!性能超越V8!

YOLOv6 3.0超越YOLOv8|更好的Neck、辅助训练分支、蒸馏学习

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
青葱年少的头像青葱年少普通用户
上一篇 2023年3月1日 下午12:26
下一篇 2023年3月1日 下午12:27

相关推荐