EDSR

Requirements

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

Name		Name	Last commit message	Last commit date
Latest commit History 10 Commits
checkpoints		checkpoints
utils		utils
README.md		README.md
calc_psnr&output.py		calc_psnr&output.py
data.py		data.py
dataset.py		dataset.py
model.py		model.py
train.py		train.py
train_pl.py		train_pl.py