
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 星标
★ 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
项目介绍
上一篇:RT-DETR
下一篇:D-FINE-seg
同类项目推荐
modlens
开源
给纯文本编码代理装上眼睛,粘贴图片即刻获得结构化视觉证据。
The first vision plugin for DeepSeek Harness, and the vision bridge for every text-o···
opencv
开源
开箱即用的视觉算法库,搞定图像视频处理与识别
Open Source Computer Vision Library
OpenCVTutorials
开源
中文 OpenCV 教程,从入门到实战,边看边跑代码
OpenCV-Python4.1 中文文档
DEIMv2
开源
用DINOv3做实时检测,又快又准,直接落地。
[DEIMv2] Real Time Object Detection Meets DINOv3