【CANN训练营第三季】进阶班AI目标属性编辑应用-StarGAN
环境搭建
申请镜像和代金券到账情况如下:
使用镜像申请ECS服务器如下:
规格 | AI加速型 | ai1s.large.4 | 2vCPUs | 8GiB | |
镜像 | | 共享镜像 版本:Ubuntu 18.04 server 64bit |
登录ECS服务器
使用“xshell”,通过SSH到ECS上(root用户):
需要修改HwHiAiUser用户的默认shell为bash。用root用户,vi /etc/passwd,修改成如下: HwHiAiUser:x:1000:1000::/home/HwHiAiUser:/bin/bash
编译运行项目
切换成HwHiAiUser前下载源代码;
wget https://sharedata.obs.cn-north-4.myhuaweicloud.com/StarGAN.zip
(下载源码的README_CN文件有说明已经不准确,samples/cplusplus/level2_simple_inference/6_other已经不存在StarGAN项目)
下载StarGAN源代码之后解压准备转换模型
sudo apt install unrar
unrar x StarGAN.rar
cd StarGAN/
下载模型并且使用atc工具转换模型
cd model/
wget https://sharedata.obs.cn-north-4.myhuaweicloud.com/StarGAN.onnx
wget https://sharedata.obs.cn-north-4.myhuaweicloud.com/aipp_nv12.cfg
atc --framework=5 --model=StarGAN.onnx --output=StarGAN_aipp --input_format=NCHW --input_shape="real_img:1,3,128,128;attr:1,5" --soc_version=Ascend310 --insert_op_conf=aipp_nv12.cfg
转换成功
从以下链接获取该样例的输入图片,放在“/data“目录下。如果目录不存在,需自行创建。
cd ../data/
wget https://sharedata.obs.cn-north-4.myhuaweicloud.com/input_1.jpg
wget https://sharedata.obs.cn-north-4.myhuaweicloud.com/input_2.jpg
编译代码。
cd ../scripts/
bash sample_build.sh
运行应用。
bash sample_run.sh