美团出品 | YOLOv6 v3.0 is coming(性能超越YOLOv7、v8)
创始人
2024-05-15 14:54:38
0次


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


一、前言简介 🎄🎈

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

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

🎄🎄YOLOv8才刚刚发布!美团出品 | YOLOv6 v3.0它来啦!其优越性能已超越当下主流检测器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,以提升小模型的推理速度。

🍀相关实验效果:

二、创新改进模块 🎄🎈

(一)BiC模块以及SimCSPSPPF模块

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

(二)Anchor-Aided Training

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

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

(三)自蒸馏策略

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

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

相关参考链接:

Beyond YOLOv8!| YOLOv6 v3.0 is coming!

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

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

相关内容

热门资讯

前端-session、jwt 目录:   (1)session (2&#x...
linux入门---制作进度条 了解缓冲区 我们首先来看看下面的操作: 我们首先创建了一个文件并在这个文件里面添加了...
关于测试,我发现了哪些新大陆 关于测试 平常也只是听说过一些关于测试的术语,但并没有使用过测试工具。偶然看到编程老师...
前缀和与对数器与二分法 1. 前缀和 假设有一个数组,我们想大量频繁的去访问L到R这个区间的和,...
nodejs:本地安装nvm实... 一、背景-使用不同版本node的原因 vue3+ts、nuxt3版本,node...
JAVA集合知识整理 Java集合知识整理 HashMap相关 HashMap的底层数据结构:jdk1.8之...
无刷直流电机介绍及单片机控制实... 无刷直流电机介绍及单片机控制实例前言基本概念优势与劣势使用寿命基本结构使用单片机控制实例电子调速器&...
fwdiary(2) dp2 1.传纸条  AcWing 275. 传纸条 - AcWing 走两条路,走一条最大的...
常用的DOS命令 常用的DOS命令 DOS(Disk Operating System,磁...
<C++> 类和对象(下) 1.const成员函数将const修饰的“成员函数”称之为const成员函数,cons...