【2023 · CANN训练营第一季】昇腾AI入门课(PyTorch)之AI应用开发入门
·
图片googlenet分类样例
- 首先在华为云上购买一台弹性云服务器,远程登陆到服务器上。
# 修改HwHiAiUser的shell为bash
vim /etc/passwd

- 切换为HwHiAiUser用户
su - HwHiAiUser

- 下载sample样例库
git clone https://gitee.com/ascend/samples.git

- 获取此应用中所需要的原始网络模型
# 进入模型目录
cd $HOME/samples/python/level2_simple_inference/1_classification/googlenet_onnx_picture/model
# 下载模型
wget https://obs-9be7.obs.cn-east-2.myhuaweicloud.com/003_Atc_Models/AE/ATC%20Model/classification/googlenet.onnx
wget https://obs-9be7.obs.cn-east-2.myhuaweicloud.com/models/googlenet_onnx_picture/insert_op.cfg

- 转换ONNX模型为离线模型
atc --model=./googlenet.onnx --framework=5 --output=googlenet_yuv --soc_version=Ascend310 --insert_op_conf=insert_op.cfg --input_shape="actual_input_1:1,3,224,224" --input_format=NCHW

-
获取样例需要的测试图片
执行以下命令,进入样例的data文件夹中,下载对应的测试图片。 cd $HOME/samples/python/level2_simple_inference/1_classification/googlenet_onnx_picture/data wget https://obs-9be7.obs.cn-east-2.myhuaweicloud.com/models/googlenet_imagenet_picture/test_image/test1.jpg
-
离线推理
cd ../src python3.6 classify.py ../data/


昇腾计算产业是基于昇腾系列(HUAWEI Ascend)处理器和基础软件构建的全栈 AI计算基础设施、行业应用及服务,https://devpress.csdn.net/organization/setting/general/146749包括昇腾系列处理器、系列硬件、CANN、AI计算框架、应用使能、开发工具链、管理运维工具、行业应用及服务等全产业链
更多推荐

所有评论(0)