Home

Awesome

ESSAformer

The implementation of ESSAformer, a network for Hyperspectral image super-resolution.

2023.10.11

The core code and trained model on Chikuseix4 are uploaded.

Enviornment

The code is tested under Pytorch 1.12.1, Python 3.8. And it might works on higer Pytorch version.

Usage

1, To use the trained model:

 checkpoint = torch.load(model_name)
 start_epoch = checkpoint["epoch"]
 net.load_state_dict(checkpoint["model"].state_dict())

2, Training
For training, just use any framework with input shape (B,C,H,W). You can use the preprocessing framework of SSPSR or MCnet for convenience.

Dataset

1,Chikusei
2,Cave
3,Pavia centre
4,Harvard
5,NTIRE2022
6,NTIRE2020
7,ICVL

To do

1,The higher order implementation of ESSA.(We find that for tiny HSI dataset, first order is sufficient. But for larger dataset, maybe higher order ESSA will bring higher accuracy)
2,Upload trained model of other 6 datasets.