YOLOv9 (学习笔记)

1.简介
  • YOLO v9主要解决了深度网络中信息丢失的问题,它的改进主要体现在 GELAN模块PGI(可编程梯度信息)
  • 实验结果:
2.网络结构
(1)整体结构
  • 训练模型结构(比推理结构多一个辅助模块,就是左边圈起来的部分,帮助梯度传播):
  • 推理模型结构:
  • 可以看到,YOLO v9和 YOLO v8相比,把C2f模块改为了GELAN(RepNCSPELAN4) ,在训练时会使用辅助结构PGI,推理时去掉辅助结构来提高推理速度。
  • 头部输出也和YOLOv8一样。
(2)GELAN模块
  • GELAN模块是把 ELAN 的"层聚合/梯度路径规划"泛化为可插拔框架 ,模块内可换 CSP/Res/Dark 等计算单元。在轻量/速度/精度三者间取得更优权衡(实际结果是又快精度又高),缓解深层信息流衰减带来的训练困难。

  • 如上图,GELAN模块是在CSPNet和ELAN模块上的融合:

  • GELAN把原本只堆叠卷积层的ELAN,泛化成一个可以"装任意计算块"的框架,这个计算块可以是任意计算单元,比如换成普通卷积,他就变成了ELAN;也可以换成残差模块......

  • 下图是信息保真度的对比实验结果:

3.重参数化
  • 上面说了,YOLO v9和 YOLO v8相比,把C2f模块改为了RepNCSPELAN4,而RepN指的就是重参数化。和YOLOv7中的重参数化差不多。
  • 重参数化分为模型训练和推理两个过程:
  • 可以看到,训练时采用残差结构,而推理检测时换为了效果相同的3×3卷积(重参数化),这样精度不变,但速度提升了。
4.PGI(可编程梯度信息)
  • PGI是在推理结构上加上了 辅助可逆分支(d图右圈)多级信息辅助(d图左圈)
相关推荐
A.说学逗唱的Coke1 小时前
【大模型专题】向量数据库深度解析:从原理到实战,构建企业级 AI 知识检索底座
数据库·人工智能
果丁智能1 小时前
智能锁赋能网约房民宿数字化管控:身份核验+远程授权,筑牢安全防线、降本增效
网络·数据库·人工智能·安全·智能家居
V搜xhliang02461 小时前
AI智能体的数据安全与合规实践
人工智能·学习·数据分析·自动化·ai编程
PPIO派欧云1 小时前
PPIO登上贵州新闻联播,深化AI算力生态建设
人工智能
hai3152475431 小时前
一种通过空间几何转换进行软件编程计算的方式与现有计算的对比
人工智能·深度学习·数学建模·硬件架构·几何学·图论·拓扑学
猿饵块2 小时前
LibreOffice---文档制作
人工智能
硅谷秋水2 小时前
HARBOR:一个面向具身智体机器人强化学习的驾驭框架
人工智能·深度学习·机器学习·机器人
Mr..Jackey2 小时前
瑞佑 RUI Builder 图形化 UI 设计工具
arm开发·人工智能·单片机·ui·人机交互·ra8889·lcd控制芯片
霍格沃兹测试开发学社测试人社区2 小时前
Skills实战:从0到1封装一个“登录鉴权”Skill,拿来即用
人工智能