一、本文介绍
本文给大家带来的是结合最新SOTA模型YOLOv9提出的辅助可逆分支结构(也就是增加辅助训练分支,PGI),该结构改动内容非常大,首先需要修改我们的yaml文件,之后需要修改检测头因为辅助可逆分支需要使用YOLOv9的DualDDetect检测头,之后我们还需要修改损失函数的计算方式以及NMS部分,本文的内容为我根据YOLOv9进行1:1移植,全网独家首发,同时本文的内容为单辅助分支 + 主干分,后面我还会发布双辅助分支 + 主分支的结构,同时针对于小白读者,我也提供了视频讲解(因为本文的修改比较大)所以大家不用担心因为变动比较大从而不会修改的问题,视频内已经介绍的非常详细了。
欢迎大家订阅我的专栏一起学习YOLO!
目录
二、原理介绍
这张图(图3)展示了可编程梯度信息(PGI)及其相关网络架构和方法。图中展示了四种不同的网络设计:
a) PAN (Path Aggregation Network):这种网络结构主要用于改进特征融合,以提高目标检测的性能。然而,由于信息瓶颈的存在,网络中可能会丢失一些信息。
b) RevCol (Reversible Columns):这是一种旨在减少信息丢失的网络设计。它通过可逆的列结构来尝试维持信息流通不受损失,但如图中“Heavy Cost”所示,这种结构会增加计算成本。
c) 深度监督:这种方法通过在网络的多个层次中插入额外的监督信号来提高学习的效率和最终模型的性能。图中显示了通过深度监督连接的各个层。
d) 可编程梯度信息 (PGI):PGI是作者提出的一种新方法(我理解的这种方法就是在前向传播的过程中没有跳级链接),它主要由三个部分组成:
1. 主分支:用于推理的架构。
2. 辅助可逆分支:生
本文详述如何在YOLOv8中结合YOLOv9的DualDDetect检测头,创新性地引入辅助可逆分支结构(PGI),以增强模型训练。内容包括PGI原理、核心代码修改教程、损失计算、超参数调整,以及详细的视频教程。全网独家,旨在解决信息瓶颈和深度监督问题,提高目标检测性能。
订阅专栏 解锁全文
(全网独家创新,附视频讲解)&spm=1001.2101.3001.5002&articleId=136471556&d=1&t=3&u=88d0e0736ca74429a1520bb8a7bf5e9d)
874

被折叠的 条评论
为什么被折叠?



