保姆级使用教程:https://blog.csdn.net/qq_36584673/article/details/138836834
data:测试单张图像以及超分结果文件夹
datasets:数据集存放文件夹,包含训练集、验证集和测试集
Plt:Loss、PSNR、SSIM与Epoch关系曲线图存放位置
epochs:模型权重文件存放位置
dataset.py:封装数据集,h5转Tensor
draw_evaluation.py:绘制Loss、PSNR、SSIM与Epoch的关系曲线图
test.py:测试单张图像超分结果
imresize.py:python实现matlab中的imresize
train.py:训练SRCNN
models.py:SRCNN模型实现
test_benchmark.py:测试benchmarks,控制台输出平均PSNR和SSIM
utils.py:图像处理,计算指标工具集
代码详细使用说明,实现细节,请看上面的教程文章!
, 相关下载链接:
https://download.csdn.net/download/qq_36584673/89305430?utm_source=bbsseo