https://github.com/mohammadshaad/mambahsi
https://github.com/mohammadshaad/mambahsi
Last synced: over 1 year ago
JSON representation
- Host: GitHub
- URL: https://github.com/mohammadshaad/mambahsi
- Owner: mohammadshaad
- Created: 2025-03-15T18:35:52.000Z (over 1 year ago)
- Default Branch: main
- Last Pushed: 2025-03-15T19:03:20.000Z (over 1 year ago)
- Last Synced: 2025-03-15T19:34:54.692Z (over 1 year ago)
- Language: Jupyter Notebook
- Size: 0 Bytes
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# MambaHSI: Spatial-Spectral Mamba for Hyperspectral Image Classification
## 📝 Introduction
* To our best knowledge, the MambaHSI is **the first image-level hyperspectral image classification model based on SSM**, which can simultaneously model long-range interaction of whole image and integrate spatial and spectral image information.
* MambaHSI demonstrates **the great potential of Mamba to be the next-generation backbone for hyperspectral image models**.
## 🚀 Getting Started
### Installation
```sh
conda create -n MambaHSI_env python=3.9
conda activate MambaHSI_env
conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 pytorch-cuda=11.7 -c pytorch -c nvidia
pip install packaging==24.0
pip install triton==2.2.0
pip install mamba-ssm==1.2.0
pip install spectral
pip install scikit-learn==1.4.1.post1
pip install calflops==0.3.2
```
### Data Preparation
The dataset can download [Google Drive](https://drive.google.com/file/d/1d-fzMXYhpwis9o_x8hPz4uHx0z5tg7LD/view?usp=sharing) and [BaiduNetdisk](https://pan.baidu.com/s/1SqzP-Y6mbuR1PRz9uGGp1Q?pwd=8ne2).
```
data
└── UP/
├── PaviaU.mat
└── PaviaU_gt.mat
...
└── Houston/
├── Houston.mat
└── Houston_GT.mat
...
└── HanChuan/
├── WHU_Hi_HanChuan.mat
└── WHU_Hi_HanChuan_gt.mat
...
└── HongHu/
├── WHU_Hi_HongHu.npy
└── WHU_Hi_HongHu_gt.npy
```
**Training:**
```
python train_MambaHSI.py --dataset_index 0
python train_MambaHSI.py --dataset_index 1
python train_MambaHSI.py --dataset_index 2
python train_MambaHSI.py --dataset_index 3
```
## 🎖️ Main Results
Pavia University Results
Houston Results
HanChuan Results
HongHu Results
