DPU

This is a repo for our work: "Dual Prior Unfolding for Snapshot Compressive Imaging".

News

Our work has been accepted by CVPR, codes and results are coming soon (July or August).

The codes and pre-trained weights have been released. More details and instructions will be continuously updated.

Results

The simulated and real results of DPU are available here.

1. Environment Requirements

Python>=3.6
scipy
numpy
einops

2. Train:

Download the cave dataset of MST series from Baidu diskcode:fo0q or here, put the dataset into the corresponding folder "DPU/CAVE_1024_28/" as the following form:

|--CAVE_1024_28
    |--scene1.mat
    |--scene2.mat
    ：
    |--scene205.mat
    |--train_list.txt

Then run the following command

cd DPU
python Train.py

3. Test:

Download the test dataset from here, put the dataset into the corresponding folder "DPU/Test_data/" as the following form:

|--Test_data
    |--scene01.mat
    |--scene02.mat
    ：
    |--scene10.mat
    |--test_list.txt

Then run the following command

cd DPU
python Test.py

For testing pre-trained models, run the following command

python Test_pretrain.py

Finally, run 'cal_psnr_ssim.m' in Matlab to get the performance metrics.

Citation

If this repo helps you, please consider citing our work:

@InProceedings{DPU,
    author    = {Zhang, Jiancheng and Zeng, Haijin and Cao, Jiezhang and Chen, Yongyong and Yu, Dengxiu and Zhao, Yin-Ping},
    title     = {Dual Prior Unfolding for Snapshot Compressive Imaging},
    booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
    month     = {June},
    year      = {2024},
    pages     = {25742-25752}
}