超分辨率重建算法搭建(Real ESRGAN)

參考的項目地址:https://github.com/xinntao/Real-ESRGAN
在前一篇Pytorch環(huán)境搭建基礎(chǔ)上進(jìn)行這篇文章的步驟
超分辨率重建AI算法

? ? ? ?簡單來講,就是讓圖片變的更清晰,相對于曾經(jīng)火爆過的Waifu2x,它的效果又有了明顯的提升。這個算法不僅適合二次元的場景,真實場景也能覆蓋。

首先,打開我們想要運行的Real-ESRGAN項目,先看README.md文件。一個完整的項目會在README里詳細(xì)說明使用方法。我們可以看到,Real-ESRGAN這個項目支持colab,exe和本地部署三種,本文采用本地部署方法,這一方法最為穩(wěn)妥。
在項目中找到requirements.txt文件,這是完整的開源項目都會有的文件,這里詳細(xì)說明了項目所依賴的第三方庫。

? ? ? ? 從上圖可以看到這里使用的是Pytorch深度學(xué)習(xí)框架以及一些其他的第三方庫,點擊Code的Download ZIP打包下載整個工程,如果會使用git,可以直接git clone工程到本地。


3. 下載好代碼后,開始安裝依賴庫,我們首先從最難安裝的Pytorch開始。打開官網(wǎng),根據(jù)我們的情況選擇安裝方式

復(fù)制命令,然后運行。(外網(wǎng)的鏈接,翻個墻也許更好)

4. 前述步驟完成后,刪除requirements.txt文件中的Pytorch和torchvision,只用python -m pip install -r requirements.txt直接打包安裝第三方依賴庫。至此,算法環(huán)境搭建完成。

5. 算法運行
README中詳細(xì)記錄了算法的運行方法。下載訓(xùn)練好的模型,將模型放到工程目錄的experiments/pretrained_models目錄下。

下載好模型,我們可以嘗試運行命令了,此時極有可能報錯:No module named 'realesrgan.version',說明包導(dǎo)入有問題??匆幌卤镜啬夸?,找到realesrgan文件夾,打開__init__.py文件,把'from .version import'這一行注釋掉就行了。之后再運行就沒有問題了。


以墨彩環(huán)為示例,圖一是原圖,圖二是重建后的圖。


可以看到,算法效果還是很棒的。