YOLOv5改进 | 注意力篇 | CGAttention实现级联群体注意力机制 (全网首发改进)

一、本文介绍

本文给大家带来的改进机制是实现级联群体注意力机制CascadedGroupAttention,其主要思想为增强输入到注意力头的特征的多样性。与以前的自注意力不同,它为每个头提供不同的输入分割,并跨头级联输出特征。这种方法不仅减少了多头注意力中的计算冗余,而且通过增加网络深度来提升模型容量,亲测在我的25个类别的数据上,大部分的类别均有一定的涨点效果,仅有部分的类别保持不变,同时给该注意力机制含有二次创新的机会。

欢迎大家订阅我的专栏一起学习YOLO!

YOLOv5改进 | 注意力篇 | CGAttention实现级联群体注意力机制 (全网首发改进)_第1张图片

专栏目录:Y

你可能感兴趣的:(YOLOv5改进有效专栏,YOLO,目标检测,深度学习,人工智能,pytorch,python,计算机视觉)