EDSR代码链接
我是在colab里跑的代码,下面给出的是在colab里的记录
阅读目录
一、colab的使用
参考这里
二、 准备工作
2.1 上传文件
首先将项目文件上传到云端硬盘,然后在云端硬盘新建一个Google Colaboratory
2.2 环境准备
将云端硬盘文件与colab关联上
!cp -r "/content/drive/My Drive/要关联的文件名" "/content/"
三、准备数据
3.1 下载数据集
!wget --no-check-certificate \
https://cv.snu.ac.kr/research/EDSR/DIV2K.tar \
-O /content/EDSR-PyTorch-master/dataset/DIV2K.tar
3.2 切换路径
import os
os.chdir("/content/EDSR-PyTorch-master/dataset")
3.3 解压数据集
!tar xvf DIV2K.tar
这里我将数据集下载到了dataset文件夹里,在src/option.py数据集路径设置如下
parser.add_argument('--dir_data', type=str, default='../dataset',
help='dataset directory')
四、训练
4.1 切换路径
import os
os.chdir("/content/EDSR-PyTorch-master/src")
4.2 从头训练EDSR
!python main.py --model EDSR --scale 2 --save edsr_x2 --n_resblocks 32 --n_feats 256 --res_scale 0.1 --reset --chop
加上指令–chop,可以解决RuntimeError: CUDA out of memory
4.3 训练结果展示
4.3.1 训练50epoch
4.3.2 loss曲线
4.3.3 PSNR曲线
五、测试
利用保存的模型进行测试
5.1 Set5
5.1.1 测试指令
!python main.py --data_test Set5 --ext img --scale 2 --n_resblocks 32 --n_feats 256 --res_scale 0.1 --pre_train /content/EDSR-PyTorch-master/experiment/model/model_best.pt --test_only --self_ensemble --save_results
利用–ext img 但是后面这个测试指令又不行了
5.1.2 测试结果
5.1.3 效果图展示
5.2 DIV2K
5.2.1 测试指令
!python main.py --data_test DIV2K --data_range 806-810 --scale 2 --n_resblocks 32 --n_feats 256 --res_scale 0.1 --pre_train ../experiment/model/model_best.pt --test_only --self_ensemble --save_results
5.2.2 测试结果
5.2.3 效果图展示
六、结果保存
参考这里
七、目前存在的问题
- benchmark
标准测试集按照demo.sh里的指令无法进行测试 - DIV2K
想测试DIV2K_test_LR_bicubic里的图像,目前没有找到解决办法
八、参考文章
https://blog.csdn.net/qq_41155645/article/details/107565773
https://blog.csdn.net/Magic_o/article/details/107325176
https://blog.csdn.net/Magic_o/article/details/107348630
https://blog.csdn.net/yldmkx/article/details/107587034
https://blog.csdn.net/m0_38129460/article/details/89846649
https://blog.csdn.net/Gavinmiaoc/article/details/90753842