ESRGANAI 超分辨率工具

ESRGAN 是一個基于增強(qiáng)型超分辨率生成對抗網(wǎng)絡(luò)的 AI 圖像處理工具，能通過 AI 在圖像超分辨率期間生成真實紋理，實現(xiàn)圖像分辨率的提升，效果如下圖所示：

測試環(huán)境

• Python 3
• PyTorch> = 1.0（如果與 CUDA 一起安裝，CUDA 版本> = 7.5。更多詳細(xì)信息）
• Python套件： pip install numpy opencv-python

測試模型

1.克隆此github存儲庫。

git clone https://github.com/xinntao/ESRGAN
cd ESRGAN

2.將低分辨率圖像放置在./LR文件夾中。（有兩個示例圖像-狒狒和漫畫）。

3.從Google云端硬盤或百度云端硬盤下載經(jīng)過預(yù)訓(xùn)練的模型。將模型放在./models。

4.作者提供了兩種具有高感知質(zhì)量和高 PSNR 性能的模型。運行測試：提供 ESRGAN模型和 RRDB_PSNR 模型，可以配置test.py。

python test.py

5.結(jié)果在./results文件夾中。