EDSR实验记录

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 效果图展示

六、结果保存

参考这里

七、目前存在的问题

1. benchmark
   标准测试集按照demo.sh里的指令无法进行测试
2. 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