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图左圈)
相关推荐
IT_陈寒4 分钟前
Vue这个特性差点让我加班到凌晨,谁懂啊
前端·人工智能·后端
D2aZXN3FhrDa7e21222 分钟前
佛山AI智能获客系统适配性:工厂门店如何挑选工具
人工智能·美诚科技有限公司
机器之心24 分钟前
WAIC现场,这家公司让一群不同的机器人共用一个大脑
人工智能·openai
小白巨白25 分钟前
玫瑰花园管理系统:AI识病+3D可视化,一套面向中小型玫瑰种植园的数字化管理工具
css·人工智能·计算机视觉·html5
延凡科技1 小时前
多场景落地复盘:端边云架构无人机智能巡检系统设计与实践
大数据·数据结构·人工智能·科技·架构·无人机·能源
科技圈观察1 小时前
壹沓科技WAIC全球首发小沓OS与小沓KE,重塑供应链AI基础设施
人工智能·科技
CV-Climber1 小时前
检索技术的实际应用
人工智能·算法
颜酱1 小时前
01 | 骨架搭建:FastAPI + Vue 跑通第一个 SSE 流式问答
前端·人工智能·后端
龙腾AI白云1 小时前
世界模型被视为破局关键
人工智能·virtualenv·知识图谱·pygame
新知图书2 小时前
10.1 项目背景与需求分析(智能客服智能体开发)
人工智能·agent·ai agent·智能体·扣子