基于pytorch的EDSR代码复现

在官网上将EDSR代码下载后，用pycharm打开
读了readme文件，提到可以先测试图像。所以下面是先测试图像的过程

第一步：下载预训练模型

readme文件里有给出链接，将下载的预训练模型保存至experiment/test/model下

第二步，改option.py中的路径

将option.py中的文件里面带有预训练的路径改为刚刚下载好的预训练模型路径

第二行的extend是什么目前还没有研究，后期再说，也顺便改了。
测试的路径不用改，代码给出了一个鸟的测试图片，就是…//test

第三步，改main函数中的Configuration

将其中的参数行改为类似demo.sh中的脚本
比如：

#1. my test on EDSR x4 Forward: 12.75s ,saving 21.96s
python3 main.py --data_test Demo  --scale 4 --n_resblocks 32 --n_feats 256 --res_scale 0.1 --pre_train ../experiment/test/model/EDSR_x4.pt --test_only --chop --save_results --self_ensemble

拷贝的时候不需要用到python3 main.py，因为就是在main函数里面改的

第四步，运行main函数

会将测试图像以预训练模型的参数生成，并将生成的图片保存到test下的result-Demo

结果如下：