作者:迪菲赫尔曼

检测头篇 | 原创自研 | YOLOv8 更换 挤压激励增强精准头 | 附详细结构图

这篇说说 YOLOv8 检测头的事情,很多同学让我发一些检测头的改进,其实检测头的改进和其他的改进也是一回事,无非就是改模块,对特征,想发我可以随时水出来 100 篇…

不过,重要的一点需要明确:检测头的改进并非仅限于在 Detect 类中的重新封装。在 yaml 配置文件中,对 neck 部分的模块化改动同样可以被视为对检测头的优化。这一点取决于你如何构建和理解你的网络结构图。
看下面的这个结构图,分了 backboneneckhead 三部分,这三部分是怎么区分的呢,我们一般就是用 yaml 区分的,没写在 Detect 里面的模块我们都算 neck,其实没必要这样想,思维要打开一些,如果我在 neckhead 的连接处加了模块,你说这个模块应该算法成 neck 还是 head 呢?

请添加图片描述

其实,这改进算在哪里就看你怎么画图, 原理就是一样的,就是讲故事的方式不一样而已,
大家看很多论文看似讲的高级检测头,又融合这种信息,那种信息的,其实不就是拼了一下特征而已,说到底出来的张量还是那个维度,中间你随便搞,只要你涨点了࿰

lock