Xiaoluobo123456 / SSUMamba

Code of "SSUMamba: Spatial-Spectral Selective State Space Model for Hyperspectral Image Denoising"

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

SSUMamba

Code of "SSUMamba: Spatial-Spectral Selective State Space Model for Hyperspectral Image Denoising"

This code is strongly based on T3SC (https://github.com/inria-thoth/T3SC)

The data can be download at https://njusteducn-my.sharepoint.com/:f:/g/personal/ironkitty_njust_edu_cn/EopcsxbL_TFJiQYEHkr9Wd8BllWJFvplYco2arbqgOhwMA?e=eqed5n .

After you download the datasets, please modify the dir path in mamba/config/test/*.yaml

  • Please install the requriment in ./ssumamba.yaml.

  • Install PyTorch 2.1.1+cu118

    pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118
  • Install causal_conv1d and mamba

    pip install -e requirements/causal-conv1d
    pip install -e requirements/mamba

About

Code of "SSUMamba: Spatial-Spectral Selective State Space Model for Hyperspectral Image Denoising"


Languages

Language:Python 83.4%Language:Cuda 11.6%Language:C++ 4.3%Language:C 0.5%Language:Shell 0.2%