Skip to content

[TCSVT 2024] Progressive Content-aware Coded Hyperspectral Compressive Imaging

Notifications You must be signed in to change notification settings

xuanyuzhang21/PCA-CASSI

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

26 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Progressive Content-aware Coded Hyperspectral Compressive Imaging

arXiv IEEE-Xplore visitors

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

idea

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}}

About

[TCSVT 2024] Progressive Content-aware Coded Hyperspectral Compressive Imaging

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages