一、本文介绍
本文还详细探讨了如何将Mamba注意力模块应用于YoloV8,创造出Yolo-Mamba网络,该网络在全局特征捕捉能力上得到了显著提升。此外、还提供将其与YOLOv8等目标检测模型有效结合改进的方法,并对自定义数据集的训练以及半自动化标注的方法进行简单介绍,帮助力读者快速在自己的数据集上进行实验与应用。并在文末提供全部代码与数据集的百度云盘链接,可一键部署运行
二、vision mamba结构与原理
Mamba成功的关键在于S6模型,该模型为NLP任务设计,通过选择性扫描空间状态序列模型,将二次复杂度降低至线性。但由于视觉信号(如图像)的无序性,Mamba的S6模型不能直接应用,
设计了 2D-selective-scan(SS2D)模块。
如上图所示,