AOT-GAN 示例

此文档讲述使用 QAI AppBuilder Python API 利用 Qualcomm® Hexagon™ Processor (NPU) 推理 AOT-GAN 图像修复模型。

示例支持设备

设备	SoC
Fogwise® AIRbox Q900	QCS9075

安装 QAI AppBuilder

提示

1. 请根据 QAI AppBuilder 安装方法 安装 QAI AppBuilder。

2. 请根据 创建 ADSP 环境变量 配置 ADSP 环境变量。

运行示例

安装依赖

在已激活的虚拟环境中安装示例依赖：

Device

pip3 install requests tqdm qai-hub py3-wget Pillow torch torchvision opencv-python-headless

运行脚本

• 进入上游示例目录

  Device

  cd qai-appbuilder/samples

• 准备输入数据（示例自带输入文件时可直接运行；也可按脚本参数指定）

input image

• 执行推理

  Device

  python3 ComputerVision/Image_Editing/aotgan/aotgan.py --chipset 9075

• 运行结果示意

mask image

output image

提示

首次运行会通过 Qualcomm® AI Hub 自动下载模型，耗时取决于网络。也可在 samples 目录使用启动器：

python3 run_inference.py --list
python3 run_inference.py --model aotgan --args "--chipset 9075"