Ryzen AI平台专属:Real-ESRGAN 128x128-tiles-amdnpu安装与配置完全指南
【免费下载链接】realesrgan-128x128-tiles-amdnpu项目地址: https://ai.gitcode.com/hf_mirrors/amd/realesrgan-128x128-tiles-amdnpu
Real-ESRGAN 128x128-tiles-amdnpu是一款专为Ryzen AI平台优化的超分辨率模型,能够将低分辨率图像提升4倍清晰度。该模型经过INT8量化处理,可在AMD AI PC的NPU上高效运行,兼顾图像质量与处理速度,是Ryzen AI用户提升图像分辨率的理想选择。
🌟 什么是Real-ESRGAN 128x128-tiles-amdnpu
Real-ESRGAN(增强型超分辨率生成对抗网络)是一种先进的AI图像超分辨率技术,能够将低分辨率图像转换为高分辨率版本。128x128-tiles-amdnpu版本特别针对AMD Ryzen AI平台进行了优化:
- 4倍超分辨率:输入图像将被放大4倍,例如320x480的图像可提升至1280x1920
- 128x128瓦片处理:采用分块处理技术,可处理任意尺寸的图像
- INT8量化优化:模型从FP32转为INT8精度,在NPU上运行效率更高
- 专为Ryzen AI设计:优化的ONNX模型可充分利用AMD NPU的计算能力
📊 超分辨率效果展示
以下是使用该模型进行4倍超分辨率处理的实际效果对比:
| 输入图像 | 输出图像 |
|---|---|
输入320x480图像通过Real-ESRGAN模型在AMD AI PC NPU上处理后,输出1280x1920的高分辨率图像
🚀 硬件与软件要求
支持的Ryzen AI处理器
| 系列 | 代号 | 发布年份 | 支持系统 |
|---|---|---|---|
| Ryzen AI Max PRO 300 Series | Strix Halo | 2025 | Windows 11 |
| Ryzen AI PRO 300 Series | Strix Point / Krackan Point | 2025 | Windows 11 |
| Ryzen AI Max 300 Series | Strix Halo | 2025 | Windows 11 |
| Ryzen AI 300 Series | Strix Point | 2025 | Windows 11 |
必备软件
- Ryzen AI软件套件(包括NPU驱动)
- Python 3.8+环境
- Git版本控制工具
⚙️ 快速安装步骤
1. 安装Ryzen AI软件
首先按照Ryzen AI安装指南安装必要的NPU驱动和Ryzen AI软件组件,整个过程大约需要30分钟。
2. 激活Ryzen AI环境
conda activate ryzen-ai-v.v.v $Env:RYZEN_AI_INSTALLATION_PATH = 'C:/Program Files/RyzenAI/v.v.v/'注意:将命令中的
v.v.v替换为实际安装的Ryzen AI版本号,如1.7.1
3. 克隆项目仓库
git clone https://gitcode.com/hf_mirrors/amd/realesrgan-128x128-tiles-amdnpu cd realesrgan-128x128-tiles-amdnpu4. 安装依赖包
pip install -r requirements.txtrequirements.txt中包含的主要依赖项:
- numpy==1.26.*
- opencv-python==4.8.*
- torch==2.6.0
- tqdm
📝 模型架构解析
Real-ESRGAN采用了增强型的生成对抗网络架构,其核心是残差-残差密集块(RRDB)结构,移除了批归一化层以提高性能:
ESRGAN架构图,展示了Residual in Residual Dense Block (RRDB)结构
Real-ESRGAN在ESRGAN基础上进行了改进,特别优化了真实世界图像的超分辨率处理:
Real-ESRGAN架构图,采用与ESRGAN相同的生成器网络,但针对不同缩放因子进行了优化
💻 运行超分辨率推理
基本推理命令
python onnx_inference.py --onnx onnx-models/realesrgan_nchw_128x128_u8s8.onnx --input 输入图像路径 --out-dir 输出目录 --device npu参数说明
--onnx:ONNX模型文件路径,通常使用onnx-models/realesrgan_nchw_128x128_u8s8.onnx--input:输入图像路径或包含图像的目录--out-dir:输出目录,处理后的图像将保存在这里--device:运行设备,npu使用AMD NPU,cpu使用CPU
示例
处理单个图像:
python onnx_inference.py --onnx onnx-models/realesrgan_nchw_128x128_u8s8.onnx --input test_image.png --out-dir outputs --device npu处理整个目录的图像:
python onnx_inference.py --onnx onnx-models/realesrgan_nchw_128x128_u8s8.onnx --input ./images --out-dir ./sr_outputs --device npu📈 模型性能评估
定量评估指标
该模型在多个基准数据集上进行了评估,包括PSNR、MS-SSIM和FID等指标:
| 模型 | Set5 PSNR | Set5 MS-SSIM | Set5 FID |
|---|---|---|---|
| 128x128(fp32) | 23.43 | 0.9346 | 114.31 |
| 128x128(int8) | 23.99 | 0.9387 | 97.89 |
INT8量化模型在保持性能的同时,提供了更好的效率
运行评估命令
python onnx_eval.py --onnx onnx-models/realesrgan_nchw_128x128_u8s8.onnx --hq-dir 高分辨率图像目录 --lq-dir 低分辨率图像目录 --out-dir 输出目录 --device npu -clean📚 高级应用
数据集准备
如需进行模型评估或进一步训练,可以下载标准超分辨率数据集:
# 下载EDSR基准数据集 python download_edsr_benchmark.py # 下载DIV2K验证集 python download_div2k.py模型缓存
模型首次运行时会进行编译并缓存,缓存文件位于modelcachekey_realesrgan_nchw_128x128_u8s8目录,后续运行将直接使用缓存文件,加快启动速度。
🧪 实际效果对比
Real-ESRGAN在真实世界图像超分辨率方面表现出色,相比其他方法能更好地去除 artifacts并恢复纹理细节:
Real-ESRGAN与其他超分辨率方法在真实世界图像上的效果对比
📄 许可证信息
本项目采用Research-only RAIL-MS许可证,仅供研究使用。
🙏 引用
如果您在研究中使用了本项目,请引用以下论文:
@InProceedings{wang2021realesrgan, author = {Xintao Wang and Liangbin Xie and Chao Dong and Ying Shan}, title = {Real-ESRGAN: Training Real-World Blind Super-Resolution with Pure Synthetic Data}, booktitle = {International Conference on Computer Vision Workshops (ICCVW)}, date = {2021} }❓ 常见问题
Q: 运行时提示NPU设备不可用怎么办?
A: 请确保已正确安装Ryzen AI软件和NPU驱动,并且使用支持的Ryzen AI处理器。可以尝试重新安装Ryzen AI环境或更新驱动。
Q: 处理大图像时出现内存不足错误?
A: 该模型采用128x128瓦片处理,理论上可以处理任意大小的图像。如遇内存问题,可尝试关闭其他应用程序或减小输入图像尺寸。
Q: 如何提高处理速度?
A: 确保使用--device npu参数,充分利用AMD NPU的硬件加速能力。首次运行会较慢,后续运行会使用缓存加速。
【免费下载链接】realesrgan-128x128-tiles-amdnpu项目地址: https://ai.gitcode.com/hf_mirrors/amd/realesrgan-128x128-tiles-amdnpu
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考