Progressive Content-aware Coded Hyperspectral Compressive Imaging

Xuanyu Zhang, Bin Chen, Wenzhen Zou, Shuai Liu, Yongbing Zhang, Ruiqin Xiong and Jian Zhang

School of Electronic and Computer Engineering, Peking University

Accepted for publication as a Regular paper in the IEEE Transactions on Circuits and Systems for Video Technology (TCSVT).

Core Ideas

Environment

pip install -r requirements.txt

Train

Please download the CAVE dataset from TSA-Net and put it into the ./datasets directory.

Two-shot Training

python train.py --template pca --outf ./exp/pca/ --method pca

Single-shot Training

python train_singleshot.py --template pca --method rnd --outf ./exp/rnd/

Test

Two-shot sampling and reconstruction. The model checkpoint file is provided in ./checkpoints/model.pth.

python test.py --template pca --outf ./exp/pca/ --method pca --pretrained_model_path ./checkpoints/model.pth

Single-shot reconstruction. The model checkpoint file is provided in ./checkpoints/model_singleshot.pth.

python test_singleshot.py --template pca --outf ./exp/rndhrnet/ --method rnd --pretrained_model ./checkpoints/model_singleshot.pth

Citation

@article{10551290,
  author={Zhang, Xuanyu and Chen, Bin and Zou, Wenzhen and Liu, Shuai and Zhang, Yongbing and Xiong, Ruiqin and Zhang, Jian},
  journal={IEEE Transactions on Circuits and Systems for Video Technology}, 
  title={Progressive Content-Aware Coded Hyperspectral Snapshot Compressive Imaging}, 
  year={2024},
  doi={10.1109/TCSVT.2024.3409421}}