news 2026/4/21 13:21:15

YOLO系列算法改进 | C3k2改进篇 | 融合MambaOut轻量化门控架构 | 纯卷积实现超越Mamba性能的同时兼顾极致轻量化与边缘部署效率 | CVPR 2025

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
YOLO系列算法改进 | C3k2改进篇 | 融合MambaOut轻量化门控架构 | 纯卷积实现超越Mamba性能的同时兼顾极致轻量化与边缘部署效率 | CVPR 2025

0. 前言

本文介绍MambaOut轻量化门控卷积模块,并将其集成到Ultralytics最新发布的YOLOv26目标检测算法中,构建C3k2_MambaOut创新模块。MambaOut由新加坡国立大学团队提出,通过移除Mamba架构中的核心令牌混合器SSM(状态空间模型),仅保留高效的门控CNN结构,旨在验证视觉任务中Mamba架构的必要性问题。将MambaOut嵌入YOLOv26的C3k2模块中,能够在ImageNet图像分类任务中超越包含SSM的视觉Mamba模型,同时凭借其纯卷积架构在边缘设备上实现更低的推理延迟与更高的硬件友好性,尤其适用于高分辨率航拍图像中的密集小目标检测、电网巡检中的微小缺陷识别、以及移动端实时检测等对轻量化与部署效率有严苛要求的场景,在保持检测精度的同时显著提升端侧推理吞吐量并降低内存占用。

专栏链接:YOLO系列算法改进专栏链接

专栏文章:YOLO26改进系列 | 卷积篇、轻量化、注意力、损失函数、Backbone、SPPF、C2PSA、Neck、检测头全方面保姆级优化合集 | 同样适配YOLOv11改进!!!

目录

0. 前言

1. MambaOut模块简介

2. MambaOut模块原理与创新点

🧠 MambaOut模块基本原理

🎯 MambaOut模块创新点

3. 具体改进步骤

🍀🍀步骤1:创建C3k2_MambaOut.py文件

🍀🍀步骤2:tasks.py文件修改

🍀🍀步骤3:创建YAML配置文件

🍀🍀步骤4:新建train.py文件训练模型

🍀🍀步骤5:模型结构打印结果


1. MambaOut模块简介

Mamba是一种具有RNN式令牌混合器的架构,其核心是状态空间模型(SSM),最近被提出用于解决注意力机制的二次复杂度问题,并随后被应用于视觉任务。然而,与基于卷积和注意力的模型相比,Mamba在视觉任务中的表现往往不尽如人意。本文深入探讨了Mamba的本质,从概念上得出结论:Mamba非常适合具有长序列和自回归特性的任务。对于视觉任务而言,由于ImageNet上的图像分类不符合上述任一特性,我们假设该任务不需要Mamba;COCO或ADE20K上的检测和分割任务虽然也不是自回归的,但它们具有长序列特性,因此我们认为探索Mamba在这些任务中的潜力仍然值得。为了实证验证我们的假设,我们通过堆叠Mamba模块并移除其核心令牌混合器SSM,构建了一系列名为MambaOut的模型。实验结果有力地支持了我们的假设。具体而言,我们的MambaOut模型在ImageNet图像分类上超越了所有视觉Mamba模型,表明Mamba确实对该任务是不必要的。至于检测和分割任务,MambaOut无法达到最先进的视觉Mamba模型的性能,这证明了Mamba在长序列视觉任务中的潜力。

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/4/21 13:19:17

STM32 DAP 烧录报错-最终解决方法的原理和操作逻辑

STM32 DAP 烧录报错(SWD/JTAG Communication Failure / Flash Download failed)最终解决方法的原理和操作逻辑针对遇到的 SWD/JTAG Communication Failure、Flash Download failed - Target DLL has been cancelled、Flash Download failed - "Cort…

作者头像 李华
网站建设 2026/4/21 13:15:31

如何快速使用res-downloader:跨平台资源下载的完整教程

如何快速使用res-downloader:跨平台资源下载的完整教程 【免费下载链接】res-downloader 视频号、小程序、抖音、快手、小红书、直播流、m3u8、酷狗、QQ音乐等常见网络资源下载! 项目地址: https://gitcode.com/GitHub_Trending/re/res-downloader 你是否经常…

作者头像 李华
网站建设 2026/4/21 13:15:26

MATLAB人形机器人技术深度解析:从运动学到平衡控制

MATLAB人形机器人技术深度解析:从运动学到平衡控制 【免费下载链接】IntroductionToHumanoidRobotics Matlab code for a Springer book "Introduction to Humanoid Robotics" 项目地址: https://gitcode.com/gh_mirrors/in/IntroductionToHumanoidRobo…

作者头像 李华
网站建设 2026/4/21 13:13:47

美股市场API技术对接指南,涵盖实时行情、历史数据、指数信息、公司详情等功能

一、概述 StockTV API为开发者提供全面的美股市场数据接口,涵盖实时行情、历史数据、指数信息、公司详情等功能。本指南详细说明如何使用API接口获取美股相关数据,所有接口均返回标准JSON格式。 二、接入准备 1. API认证 所有接口请求需在URL参数中包含有…

作者头像 李华
网站建设 2026/4/21 13:11:35

OpenCV 4.5.5实战:用calcOpticalFlowFarneback搞定图像对齐(附参数调优心得)

OpenCV 4.5.5实战:calcOpticalFlowFarneback图像对齐参数调优全解析 在计算机视觉领域,图像对齐是一个基础但至关重要的任务。无论是文档扫描件的自动校正、医学影像的配准,还是多视角图像的拼接,都需要精确的对齐技术作为支撑。而…

作者头像 李华