mengkunzhao / HSI_baseline Public

forked from Jiamian-Wang/HSI_baseline

Notifications You must be signed in to change notification settings
Fork 0
Star 0

A Simple and Efficient Reconstruction Backbone for Snapshot Compressive Imaging

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 14 Commits
.idea		.idea
Data		Data
__pycache__		__pycache__
architecture		architecture
models		models
README.md		README.md
common.py		common.py
cover_figure-page1v2.png		cover_figure-page1v2.png
cover_figure.png		cover_figure.png
framework_v4.png		framework_v4.png
models_resblock_v2.py		models_resblock_v2.py
models_resblock_v3.py		models_resblock_v3.py
models_resblock_v4.py		models_resblock_v4.py
ssim_torch.py		ssim_torch.py
test.py		test.py
utils.py		utils.py

Repository files navigation

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] (https://github.com/Jiamian-Wang/HSI_baseline/tree/main/Data/testing/simu). Make sure to specify the test_path before evaluation. The coded apture can be accessed [here] (https://github.com/Jiamian-Wang/HSI_baseline/tree/main/Data). 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] (https://github.com/Jiamian-Wang/HSI_baseline/tree/main/models/v1) 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.