
基于Pytorch的图像超分辨率RCAN复现代码及科研绘图,包含指标计算与最佳SSIM和PSNR模型权重(x2、x3、x4、x8)
5星
- 浏览量: 0
- 大小:None
- 文件类型:ZIP
简介:
本项目基于PyTorch实现图像超分辨率技术中的RCAN算法,并提供科研绘图、性能指标(如SSIM, PSNR)评估及不同倍率预训练模型的下载。
配套文章介绍了以下内容:
- 测试单张图像以及超分结果文件夹:存放测试过程中生成的超分辨率图片。
- 数据集存放文件夹(datasets):包含训练数据和测试数据,用于模型训练与验证。
- Plt:保存Loss、PSNR、SSIM指标随Epoch变化曲线图的位置。
- 权重文件存放位置(weights):存储训练好的RCAN模型权重参数。
此外还包括:
- data_aug.py:实现离线数据增强功能;
- dataset.py:负责生成训练集和验证集,其中训练集中包含在线数据增强;
- draw_evaluation.py:绘制Loss、PSNR、SSIM与Epoch的关系曲线图;
- example.py:用于测试单张图像的超分辨率效果;
- imresize.py:实现MATLAB中的imresize函数功能;
- main.py:执行RCAN模型的训练过程;
- model.py:定义并实现了RCAN模型架构;
- save_benchmark_sr.py:将基准数据集上的SR结果保存下来;
- test_benchmark.py:测试5个标准基准,输出平均PSNR和SSIM值。
具体使用方法请参考配套文章。
全部评论 (0)
还没有任何评论哟~


