【图像超分】论文复现:Pytorch实现RDN!保姆级复现教程!实现与原论文基本一致的PSNR和SSIM!代码注释详尽!易读易复用!可用于训练自己的数据集!附完整代码和各放大倍数下的最优模型权重文件!
论文题目:Residual Dense Network for Image Super-Resolution —— 用于图像超分辨率的残差密集网络【图像超分】论文精读:Residual Dense Network for Image Super-Resolution(RDN)请配合上述论文精读文章使用,效果更佳!虽然论文中为了丰富内容用了很多缩写,并且写的很详细。这对于论文复现者来说是很友好的,尤其是网络结构的实现上,非常清晰。
【图像超分】论文复现:Pytorch实现RDN!保姆级复现教程!实现与原论文基本一致的PSNR和SSIM!代码注释详尽!易读易复用!可用于训练自己的数据集!附完整代码和各放大倍数下的最优模型权重文件!

第一次来请先看这篇文章:【超分辨率(Super-Resolution)】关于【超分辨率重建】专栏的相关说明,包含专栏简介、专栏亮点、适配人群、相关说明、阅读顺序、超分理解、实现流程、研究方向、论文代码数据集汇总等)

项目代码和最优性能的权重文件下载见文末链接!!!可以复现论文、训练自己的数据集,详细代码使用说明!!!同时包含最优性能的模型权重文件(x2、x3、x4),可以直接用来超分自己的图像!!!

本文亮点:

  • 详解RDN的网络结构、训练流程、测试流程、画图流程,代码注释详细,无论是科研还是应用,新手小白都能看懂,学习阅读毫无压力
  • 提供完整代码和x2、x3、x4训练好的最优模型权重文件,可以直接拿来超分自己的图像数据;
  • 只需简单替换训练集,即可训练你自己的图像数据集,给出了可能的调参方式
  • 量化评估实现,计算测试集Set5、Set14、BSD100、URBAN100、MANGA109的平均PSNR和SSIM,指标与原论文基本一致,有的甚至更高
  • 质量评估实现,可视化图像超分结果