Joint Optimization of Splitter Pattern and Image Reconstruction for Metasurface-based Color Imaging Systems
# create and activate new conda environment
conda create -n splitter python=3.10
conda activate splitter
# install pytorch (version: 2.1.1)
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
# install required packages
pip install -r requirements.txt
# clone the repository
git clone https://github.com/haosennn/Splitter.git
cd ./Splitter/
Training and test datasets can be downloaded from the following websites:
- DIV2K: https://data.vision.ee.ethz.ch/cvl/DIV2K/
- Kodak24: https://r0k.us/graphics/kodak/
- McMaster: https://www4.comp.polyu.edu.hk/~cslzhang/CDM_Dataset.htm
Pretrained models can be downloaded from: https://drive.google.com/drive/folders/1F_cTzA7KaeKz6eEpiT-CEuQozV7WtRL9?usp=drive_link
python main_test.py --exp 1 --save --sigmas 0 --data_test Kodak24
python main_train.py --S_type A --gpu_ids 0