# Mamba-YOLO

**Repository Path**: fork-project/Mamba-YOLO

## Basic Information

- **Project Name**: Mamba-YOLO
- **Description**: No description available
- **Primary Language**: Unknown
- **License**: Apache-2.0
- **Default Branch**: main
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2024-07-19
- **Last Updated**: 2024-07-19

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# Mamba-YOLO
Official pytorch implementation of “Mamba-YOLO：SSMs-based for Object Detection”

![Python 3.11](https://img.shields.io/badge/python-3.11-g)
![pytorch 2.3.0](https://img.shields.io/badge/pytorch-2.3.0-blue.svg)
[![docs](https://img.shields.io/badge/docs-latest-blue)](README.md)

![](asserts/SOTACompare.png)

## Installation
``` shell
# pip install required packages
conda create -n mambayolo -y python=3.11
conda activate mambayolo
pip3 install torch===2.3.0 torchvision torchaudio
pip install seaborn thop timm einops
cd selective_scan && pip install . && cd ..
pip install -v -e .
```

## Training

```shell
python mbyolo_train.py --task train --data ultralytics/cfg/datasets/coco.yaml \
 --config ultralytics/cfg/models/v8/mamba-yolo.yaml \
--amp  --project ./output_dir/mscoco --name mambayolo_n
```

## Acknowledgement

This repo is modified from open source real-time object detection codebase [Ultralytics](https://github.com/ultralytics/ultralytics). The selective-scan from [VMamba](https://github.com/MzeroMiko/VMamba).