Official implementation of the following paper
Yutong Liu, Zhen Cheng, Zeyu Xiao, and Zhiwei Xiong, Light Field Super-Resolution Using Decoupled Selective Matching
- Python 3 (Recommend to use Anaconda)
- Pytorch 1.7.0
- einops
- Numpy
- Scipy
- matplotlib
- TensorboardX
- MATLAB (For data preparation)
- We make experiments on two benchmarks CiytU and BasicLFSR.
- For the benchmark CiytU, please refer to ATO or SAV_conv for the preparetion of the dataset. You can downland the test dataset from BaiduYun and put them into the folder ./CiytU/data/ for a readily start.
- For the benchmark BasicLFSR, please refer to BasicLFSR for the preparetion of the dataset. You can downland the test dataset from BaiduYun and put them into the folder ./BasicLFSR/data/ for a readily start.
- For the Pretrained Model, please downland checkpoint from BaiduYun and put them into the folder ./CiytU/pretrained_model/, while please downland checkpoint from BaiduYun and put them into the folder ./BasicLFSR/pretrained_model/.
For CityU, to train and test our DSMNet under the scale of 2 as an example:
cd ./CityU/
bash train_CityU_scale2.sh
bash test_CityU_scale2.sh
For BasicLFSR, to train and test our DSMNet under the scale of 2 as an example:
cd ./BasicLFSR/
bash train_BasicLFSR_scale2.sh
bash test_BasicLFSR_scale2.sh
If you find this work helpful, please consider citing our paper.
@ARTICLE{10268449,
author={Liu, Yutong and Cheng, Zhen and Xiao, Zeyu and Xiong, Zhiwei},
journal={IEEE Transactions on Circuits and Systems for Video Technology},
title={Light Field Super-Resolution Using Decoupled Selective Matching},
year={2023},
volume={},
number={},
pages={1-1},
doi={10.1109/TCSVT.2023.3321085}}
If you have any problem about the released code, please contact me with email (ustclyt@mail.ustc.edu.cn).