代码拉取完成，页面将自动刷新
中文|English
此链接提供Vgg16 TensorFlow模型在NPU上离线推理的脚本和方法
VAEGAN 推理，基于 VAE/GAN
此案例仅为您学习Ascend软件栈提供参考，不用于商业目的。
在开始之前，请注意以下适配条件。如果不匹配，可能导致运行失败。
| Conditions | Need |
|---|---|
| CANN版本 | >=5.0.3 |
| 芯片平台 | Ascend310/Ascend310P3 |
| 第三方依赖 | 请参考 'requirements.txt' |
git clone https://gitee.com/ascend/ModelZoo-TensorFlow.git
cd Modelzoo-TensorFlow/ACL_TensorFlow/contrib/cv/VAEGAN_for_ACL
请自行下载数据集, 更多详细信息请参见: CelebA
获取ckpt模型, 更多详细信息请参见: ckpt
构建推理应用程序, 更多详细信息请参见: xacl_fmk
环境变量设置 请参考说明，设置环境变量
预处理
python3 main.py --act dump
Pb模型转换为om模型
atc --model=./model/VAE_GAN_gpu.pb --framework=3 --output=./model/VAE_GAN_gpu --soc_version=Ascend310 --input_shape="Placeholder:64,64,64,3" --log=info
运行推理
python3 inference/xacl_inference.py
后期处理
python3 main.py --act compare
我们通过PSNR来衡量性能。
| ascend310 | |
|---|---|
| PSNR | 11.922089 |
此处可能存在不合适展示的内容，页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容，可点击提交进行申诉，我们将尽快为您处理。
马建仓 AI 助手