
Matlab代码解释 - SRCNN:论文《学习深度卷积网络以实现图像超分辨率》(ECCV20...),其中使用了PyTorch和Matlab。
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
该项目是对Pytorch实现的《学习深度卷积网络以实现图像超分辨率》(ECCV2014)论文的直接复制。为了便于理解和应用,项目依赖于Matlab 2016和火炬1.0.0工具包,其中包含了论文作者提供的相关Matlab代码。选择同时使用Matlab和Python主要源于两种语言在双三次插值方法上的差异,这些差异在使用PSNR指标评估时会导致结果的显著偏差。 项目的概述如下:首先,通过执行./data_pro/generate_train.m脚本生成用于训练的train.h5文件。随后,使用./data_pro/generate_test.m脚本生成用于测试的test.h5文件。接下来,运行train.py脚本,将Pytorch模型.pkl转换为Matlab矩阵格式的.mat文件(weights.pkl转化为weights.mat)。最后,执行pythonconvert.py脚本,利用./test_link/get_result.m脚本获取PSNR指标结果并重建RGB图像。最终结果可以通过./model/weights.mat文件获得:在Set5数据集上的平均重建PSNR值为32.44dB,而使用双三次插值的平均PSNR值为30.39dB;在Set14数据集上的平均重建PSNR值也呈现类似趋势。
全部评论 (0)
还没有任何评论哟~


