Mamba-YOLO

用 Mamba 替 YOLO 主干,精度更高算力更省

Mamba-YOLO 是论文《Mamba-YOLO: SSMs-based for Object Detection》的官方 PyTorch 实现,把状态空间模型(SSM/Mamba)引入目标检测,替代 YOLO 系列常用的 CNN 或 Transformer 主干。它要解决的问题是:Transformer 检测器精度高但自注意力计算量随分辨率平方增长,CNN 又受限于局部感受野。项目用 Mamba 的线性复杂度序列建模能力,在保持全局信息交互的同时降低计算开销,并给出适配检测任务的网络结构与训练配置。核心能力包括可复现的检测精度、对 COCO 等标准数据集的训练/评估脚本,以及与 YOLO 框架相近的工程接口,方便研究者对比 SSM 与 CNN/Transformer 主干的差异。

开源 free 计算机视觉
访问官网 ↗ GitHub ↗ 文档 ↗
GitHub 星标 ★ 655
维护状态 较活跃
是否开源
定价模式 free

项目数据

分类计算机视觉
开发团队HZAI-ZJNU
所属国家
官网地址
定价模式free
价格说明开源免费,需自行部署
访问状态
是否开源
开源协议AGPL-3.0
主要语言Python
技术栈/模型
GitHub 星标★ 655
30天Star增速
HF 下载量
上线时间2024-06-09 00:00:00
最近更新2026-09-17 00:00:00
维护状态较活跃
中文支持
访问方式
移动端支持
综合评分
收录时间2026-09-20
浏览次数0

使用教程

核心亮点

  • 把 Mamba/SSM 落到目标检测任务,提供官方 PyTorch 训练与评估代码,可直接复现论文结果
  • 相比 Transformer 检测器,序列建模为线性复杂度,高分辨率特征图上计算更友好
  • 沿用 YOLO 式工程结构,权重、配置、数据接口清晰,便于替换主干做消融实验

不足之处

  • 依赖 mamba-ssm、causal-conv1d 等 CUDA 扩展,环境安装对显卡和编译工具链较敏感
  • 代码与文档主要面向研究复现,缺少部署、量化和移动端推理支持

适用场景

  • 研究 SSM 与 CNN/Transformer 在检测任务上的精度-速度权衡
  • 在边缘或算力受限设备上探索线性复杂度检测主干
  • 以 YOLO 框架为基础,替换主干做目标检测消融与改进实验

替代项目

YOLOv8、RT-DETR、VMamba

项目介绍

Mamba-YOLO 是一个计算机视觉领域的开源项目,官方简介:the official pytorch implementation of “Mamba-YOLO:SSMs-based for Object Detection”。项目使用 Python 开发,在 GitHub 上获得 655 星标。

上一篇:RT-DETR

下一篇:D-FINE-seg

同类项目推荐

modlens 开源

给纯文本编码代理装上眼睛,粘贴图片即刻获得结构化视觉证据。

The first vision plugin for DeepSeek Harness, and the vision bridge for every text-o···

★ 3995 2026-08-19
opencv 开源

开箱即用的视觉算法库,搞定图像视频处理与识别

Open Source Computer Vision Library

★ 90893 2026-08-10
OpenCVTutorials 开源

中文 OpenCV 教程,从入门到实战,边看边跑代码

OpenCV-Python4.1 中文文档

★ 1430 2026-08-20
DEIMv2 开源

用DINOv3做实时检测,又快又准,直接落地。

[DEIMv2] Real Time Object Detection Meets DINOv3

★ 2035 2026-08-14