【CANN训练营第三季】进阶班AI目标属性编辑应用-StarGAN

【CANN训练营第三季】进阶班AI目标属性编辑应用-StarGAN

环境搭建

申请镜像和代金券到账情况如下：

 

使用镜像申请ECS服务器如下：

规格	AI加速型 | ai1s.large.4 | 2vCPUs | 8GiB
	镜像	6.0.RC1.alpha001_new  | 共享镜像 版本：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