PS_Mamba

Project Layout

.
|-- data/
|-- images/
|-- model/
|-- utils/
|-- calflops/
|-- mamba_ssm/
|-- README.md
`-- train_PSMamba.py

Installation

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

Data Preparation

Place the datasets under data/ like this:

data/
|-- UP/
|   |-- PaviaU.mat
|   `-- PaviaU_gt.mat
|-- HongHu/
|   |-- WHU_Hi_HongHu.mat
|   `-- WHU_Hi_HongHu_gt.mat
`-- KSC/
    |-- KSC.mat
    `-- KSC_gt.mat

Training

python train_PSMamba.py --dataset_index 0
python train_PSMamba.py --dataset_index 1
python train_PSMamba.py --dataset_index 2

Dataset indices:

• 0: UP
• 1: HongHu
• 2: KSC