0
点赞
收藏
分享

微信扫一扫

【2023 · CANN训练营第一季】昇腾AI入门课(PyTorch)之AI应用开发入门

图片googlenet分类样例

  1. 首先在华为云上购买一台弹性云服务器,远程登陆到服务器上。

# 修改HwHiAiUser的shell为bash
vim /etc/passwd

  1. 切换为HwHiAiUser用户

su - HwHiAiUser

  1. 下载sample样例库

git clone https://gitee.com/ascend/samples.git

  1. 获取此应用中所需要的原始网络模型

# 进入模型目录
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

  1. 转换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

  1. 获取样例需要的测试图片

执行以下命令,进入样例的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

  1. 离线推理

cd ../src
python3.6 classify.py ../data/

【2023 · CANN训练营第一季】昇腾AI入门课(PyTorch)之AI应用开发入门_昇腾CANN

举报

相关推荐

0 条评论