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

143 阅读1分钟

【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 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 sharedata.obs.cn-north-4.myhuaweicloud.com/StarGAN.onn…

    wget sharedata.obs.cn-north-4.myhuaweicloud.com/aipp_nv12.c…

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 sharedata.obs.cn-north-4.myhuaweicloud.com/input_1.jpg

    wget sharedata.obs.cn-north-4.myhuaweicloud.com/input_2.jpg

 

编译代码。

cd ../scripts/

    bash sample_build.sh

   

运行应用。

    bash sample_run.sh