Skip to content

ZhaoHengrun/VDSR

Repository files navigation

VDSR

一个VDSR的pytorch复现,在原版基础上瞎搞,没有彻底完善,仅仅是能跑。使用adam优化器,尝试了不同的学习率调整策略(但是没一个靠谱的),将网络输入的图像块大小改为256*256。
参考自https://github.com/pytorch/examples/tree/master/super_resolution
https://github.com/twtygqyy/pytorch-vdsr

Requirements

Python 3.8
PyTorch 1.6.0
Numpy 1.19.2
Pillow 7.2.0

Usage:

Make datasets

使用make_datasets/目录下的make_dataset.m制作数据集,训练,验证,测试的图片分别保存在datasets/train,datasets/valid,datasets/test目录下

Train

运行python main.py进行训练
模型保存在checkpoints/目录下

Test

运行python calc_psnr&output.py进行测试,生成的图片保存在results/目录下

Eval

calc_psnr&output.py中写的psnr计算公式可能有误差,也可使用make_datasets/目录下的compute_psnr.m计算psnr

About

小白写的第一个复现

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages