EDSR Baseline的pytorch复现
参考自
https://arxiv.org/pdf/1707.02921.pdf
https://github.com/twtygqyy/pytorch-edsr
https://github.com/thstkdgus35/EDSR-PyTorch
Python 3.8
PyTorch 1.6.0
Numpy 1.19.2
Pillow 7.2.0
Visdom 0.1.8.9
Wandb 0.10.10
所用的目录需要手动创建
使用utils/目录下的generate_train.m制作训练数据集
使用utils/目录下的make_dataset.m制作测试数据集
生成的训练，测试数据集图片分别保存在datasets/train，datasets/test目录下
运行python train.py进行训练
模型保存在checkpoints/目录下
运行python calc_psnr&output.py进行测试，生成的图片保存在results/目录下
calc_psnr&output.py中写的psnr计算结果可能有问题，可使用utils/目录下的compute_psnr.m计算psnr