1.改动
- 在原来模型的基础上增加深度。
- 替换训练数据集。由原来的VOC2012替换为DIV2K(共800张)。如下:
target_pic:( 2040 × 1644 2040 \times 1644 2040×1644)
data_pic:( 1020 × 822 1020 \times 822 1020×822,双三次插值)
备注:
1.没有对训练集数据进行统一裁剪处理,各个图片大小不一。batchsize=1
2.验证集和测试集未作改变。
2.效果
2.1 原模型+新训练集
训练集效果:50个epochs,psnr ≈ 30.1 \approx 30.1 ≈30.1
测试集效果:
target:
result:(new)
result:(old)
psnr ≈ 26.6 \approx 26.6 ≈26.6
在训练集上表现好并不能说明一切,测试集上表现好才是王道。
2.2 新模型+新训练集
网络容易挂掉。
有待修改;应先用旧数据集跑通。