SRCNN论文的代码复现-超分辨率在神经网络的开山之作

代码地址：https://github.com/se16wangmeiwei/SRCNN-pytorch/tree/code/srcnn
论文地址：https://arxiv.org/abs/2010.02414

我是刚刚跨入图像超分辨率方向的小白，看了SRCNN的论文，然后进行了复现。

数据集的预处理
我的数据集是先下载到本地上再进行读取的。
data.py
dataset.py

设定模型训练类
demo.py

模型的设置
model.py

测试模块
predict.py

最终的训练结果
训练结果不是很好，论文中的PSNR是32，我是23。应该是需要调参数的。还有一个原因导致结果不好，应该是训练的图片太少了。

最终的测试效果：
很明显，只是简单的将一些马赛克变得比较平滑了而已。

最后，很少写博，可能写得不尽人意，哪里写得不好的希望大家指出进行改正。大家有什么问题也可以在评论区中提出。抱拳。
【参考了一个博主的代码，但是找不到了…如果那个博主看到了可以给我说下。】