配套文章:https://blog.csdn.net/qq_36584673/article/details/136861864
文件说明:
benchmark_results:保存不同倍数下测试集的测试结果
data:存放数据集的文件夹,包含训练集、测试集、自己的图像/视频
epochs:保存训练过程中每个epoch的模型文件
statistics:存放训练和测试的评估指标结果
training_results:存放每一轮验证集的超分结果对比,每张图像5行3列展示
data_utils.py:数据预处理和制作数据集
demo.py:任意图像展示GT、Bicubic、SRGAN可视化对比结果
draw_evaluation.py:绘制Epoch与Loss、PSNR、SSIM关系的曲线图
loss.py:损失函数
model.py:网络结构
test_benchmark.py:生成benchmark测试集结果
test_image.py:生成任意单张图像用SRGAN超分的结果
test_video.py:生成SRGAN视频超分的结果
train.py:训练SRGAN
使用方法见文章。
1