ESRGANAI 超分辨率工具
ESRGAN 是一个基于增强型超分辨率生成对抗网络的 AI 图像处理工具,能通过 AI 在图像超分辨率期间生成真实纹理,实现图像分辨率的提升,效果如下图所示:
测试环境
- Python 3
- PyTorch> = 1.0(如果与 CUDA 一起安装,CUDA 版本> = 7.5。更多详细信息)
- Python套件:
pip install numpy opencv-python
测试模型
1.克隆此github存储库。
git clone https://github.com/xinntao/ESRGAN cd ESRGAN
2.将低分辨率图像放置在./LR
文件夹中。(有两个示例图像-狒狒和漫画)。
3.从Google云端硬盘或百度云端硬盘下载经过预训练的模型。将模型放在./models
。
4.作者提供了两种具有高感知质量和高 PSNR 性能的模型。运行测试:提供 ESRGAN模型和 RRDB_PSNR 模型,可以配置test.py
。
python test.py
5.结果在./results
文件夹中。
评论