【2024·CANN训练营第一季】从图片分类入门推理应用开发
收藏回复举报
【2024·CANN训练营第一季】从图片分类入门推理应用开发
发表于2024-03-21 14:50:01
0 查看

1. 安装驱动固件和CANN环境

参考昇腾文档安装驱动固件和CANN环境

安装驱动和固件-软件安装-环境准备-...-文档首页-昇腾社区 (hiascend.com)

2. 安装完驱动固件后,npu-smi info查看NPU信息

cke_2193.png

3. 安装完ascend-toolkit后,查看版本

source /usr/local/Ascend/ascend-toolkit/set_env.sh配置环境变量

cke_48139.png

 

4. 从gitee上克隆Ascend的samples仓库到本地

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

 cke_30626.png

5. 进入samples/inference/modelInference/sampleResnetQuickStart/cpp目录下

cke_312.png

 

6. 安装opencv

apt install libopencv-dev

cke_313.png

 

7. 设置opencv2软链接

ln -s /usr/include/opencv4/opencv2 /usr/include/opencv2

 

8. 下载resnet50.onnx到model目录

cd model/ && wget wget https://obs-9be7.obs.cn-east-2.myhuaweicloud.com/003_Atc_Models/resnet50/resnet50.onnx

cke_314.png

 

9. atc模型转换

atc --model=resnet50.onnx --framework=5 --output=resnet50 --input_shape="actual_input_1:1,3,224,224"  --soc_version=Ascend310

cke_315.png

 

10. 获取测试图片

cd ../data/ && wget https://obs-9be7.obs.cn-east-2.myhuaweicloud.com/models/aclsample/dog1_1024_683.jpg

cke_316.png

 

11. 配置程序编译依赖的头文件与库文件路径,并编译样例源码

cke_317.png

 

12. 模型推理bash sample_run.sh

cke_318.png

可视化结果

cke_12623.png

本帖最后由 匿名用户2024/04/08 11:26:20 编辑

我要发帖子