new baseline for CASSI measurement reconstruction

This repository contains the test code and pre-trained models for the arXiv paper "******".

Data

The benchmark simulation test dataset can be accessed here. Make sure to specify the test_path before evaluation. The coded apture can be accessed here. Make sure to specify the mask_path before evaluation.

Three versions of proposed method

Checkpoint of model v1: w/o rescaling pairs can be accessed here Similarly, pre-trained models of the other two versions are also been released. Makre sure to specify the model_path as desired pre-trained model.

Citation

If you find new baseline useful, please cite the following paper

******

For any questions, please submit an issue or contact jiamiansc@gamil.com.