从零开始Real-ESRGAN的复现

前言

        要初步了解Real-ESRGAN，可以看我之前发布的博客。初学Real-Esrgan-CSDN博客

        本文主要是对Real-ESRGAN的一个复现，主要就是对环境的配置进行记录，因为是实现之后才做的记录，所以只能尽可能的对其进行复现。大家可以看一下这个博主的文章进行一个参考。【论文阅读+测试】Real-Esrgan超分辨率算法_real esrgan-CSDN博客

项目开始

        首先，我们要知道Real-ESRGAN这个项目是一个pytorch项目（需要配环境，可以自己训练、微调），如果是第一次配置环境，那么就需要安装Python环境（因为笔记本电脑配置比较低，所以我是在服务器上面配置的环境，我这使用的是miniconda环境，这个在网上也挺多资料的，大家可以在网上自行查询下载）。

下载安装完环境之后，使用conda创建虚拟环境。

conda create -n py38_esrgan python=3.8         #py38_esrgan可以换成自己环境的名字

安装pytorch(在前期安装的时候可以先随意安装一个版本，后面安装其他包需要啥版本再改成其他版本)

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1

根据项目安装所需要的包，在这里主要有：basicsr、facexlib、gfpgan。如果之前配置过这个项目，那应该知道这个项目比较难配的就是这个basicsr包，这里先把安装的命令写下来。

pip install basicsr
pip install facexlib
pip install gfpgan
pip install -r requirements.txt

对于这个basicsr包，有可能是torch的问题，也有可能是其他包的问题，所以解决这个问题的方法就是以安装这个包为最高优先级，他需要什么版本就安装什么版本。但是还有一个需要注意的是，也有可能是下载源的问题，我一开始使用的是清华源，一直都报错后面换阿里源之后一下就安装好了。在这参考的是无法安装basicsr的解决方法_basicsr安装失败-CSDN博客。

把basicsr安装完之后应该剩下的都不是什么大问题了。然后就可以进行测试了。

#--face_enhance可以不用，这个是人脸增强的权重文件，需要额外下载的
python inference_realesrgan.py -n RealESRGAN_x4plus -i inputs --face_enhance

 再运行这个命令的时候会缺几个权重文件，我的建议就是缺啥就下啥（这些文件我都是本机下载然后放入服务器里面的，本机下载的速度快一点），他也会把文件存放的目录显示出来，下好之后把文件放到正确的位置上就好了。

把权重文件下载好之后就可以直接运行出结果，输入图片放在inputs文件夹里面，输出结果会被保存在result文件夹里面。

训练

如果想要自己训练模型可以参考：Real-ESRGAN/docs/Training_CN.md at master · xinntao/Real-ESRGAN (github.com)

总结一下的话那就是主要修改两个文件

options/train_realesrnet_x4plus.yml

options/train_realesrgan_x4plus.yml

 修改啥内容就看着参考连接里面看着修改就行。需要注意的是训练这个模型需要的内存比较大。