第一次来请先看这篇文章:【超分辨率(Super-Resolution)】关于【超分辨率重建】专栏的相关说明,包含专栏简介、专栏亮点、适配人群、相关说明、阅读顺序、超分理解、实现流程、研究方向、论文代码数据集汇总等)
代码和模型权重文件下载见文末链接!!!包含最优性能权重文件(x2、x4、x8),可直接用于测试和超分自己的图像或视频。
本文亮点:
- 详解SRGAN的网络结构和训练流程,代码注释详细,新手小白都能看懂。
- 与前几篇基于CNN的复现相比,增加了用验证集验证模型的过程,可以通过控制台实时查看损失、PSNR和SSIM的变化,以便发现不对后及时调整。
- 可以测试主流的测试集、超分自己的图像数据、超分自己的视频数据,生成对比结果。
- 与源码相比,新增画图代码(Epoch与Loss、PSNR、SSIM的关系曲线),方便查看训练趋势以便改进,实现科研作图。
- 此外,还新增了与Bicubic对比的超分结果可视化代码。
- 详述代码可以改进的部分,以更好训练你自己的数据集。
- 提供训练好的模型权重文件(x2,x4,x8均有),可以拿过来直接用。