Home

Awesome

DPHSIR

Paper | Pretrained Model

Deep Plug-and-Play Prior for Hyperspectral Image Restoration (Neurocomputing 2022)

Zeqiang Lai, Kaixuan Wei, Ying Fu

<img src="algorithm.png" width="350" />

News :sparkles:

Usages

  1. Install the requirments
# Pytorch >= 1.8
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
conda install -c conda-forge opencv
pip install -r requirements.txt
  1. Clone the repo
git clone https://github.com/Zeqiang-Lai/DPHSIR.git
cd DPHSIR
pip install -e .
  1. Run cli or playgrounds
# run cli
python cli/main.py -i [input_path] [task]
# run playground
python playgrounds/deblur.py

Citation

If you find our work useful for your research, please consider citing our paper :)

@article{lai2022dphsir,
    title = {Deep plug-and-play prior for hyperspectral image restoration},
    journal = {Neurocomputing},
    volume = {481},
    pages = {281-293},
    year = {2022},
    issn = {0925-2312},
    doi = {https://doi.org/10.1016/j.neucom.2022.01.057},
    author = {Zeqiang Lai and Kaixuan Wei and Ying Fu},
}

Acknowledgement