WDSR是在EDSR代码的基础上实现的,代码链接在这里
一、准备过程
1.1 代码修改
参考这里
1.2 操作准备
参考这里
二、训练
2.1 训练命令
!python main.py --model wdsr_b --epochs 50 --scale 2 --save wdsr_b_x2 --n_resblocks 32 --n_feats 128 --res_scale 0.1
2.2 训练结果
2.3 loss曲线
2.4 PSNR曲线
三、测试
3.1 测试命令
!python main.py --model wdsr_b --data_test DIV2K --data_range 806-810 --scale 2 --n_resblocks 32 --n_feats 128 --res_scale 0.1 --pre_train ../experiment/wdsr_b_x2/model/wdsr_model_best.pt --test_only --self_ensemble --save_results
3.2 测试结果
3.3 效果图展示