图片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/
标签:入门,AI,onnx,样例,googlenet,CANN,--,com,obs
From: https://blog.51cto.com/u_14125503/6378084