推理模型应用的架构如下

在这里插入图片描述

安装samples

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

获取mnist的air格式模型

wget https://modelzoo-train-atc.obs.cn-north-4.myhuaweicloud.com/003_Atc_Models/AE/ATC%20Model/lenet/mnist.air

模型转换

ATC可以实现模型转换
在这里插入图片描述
运行命令将air模型转为昇腾310推理芯片可以运行的om模型

atc --framework=1 --model=mnist.air --output=mnist --soc_version=Ascend310

这个时候我那边报错了,在此记录了一下:

acl报错及其解决方法(这一步卡了我半天时间,重点记录)

在这里插入图片描述
昇腾社区的200DK教程下载的PyACl的acl版本可能不匹配,我们需要根据我们的CANN版本来下载对应的CANN包,CANN包里面有PyACL的run包。我们以Ascend-cann-toolkit_5.0.4.alpha005_linux-aarch64为例,
去官网下载:https://www.hiascend.com/software/cann/community-history

在这里插入图片描述

解压Ascend-cann-toolkit_5.0.4.alpha005_linux-aarch64.run后可以使用pyACL,scp上传到我们的开发套件中。
在这里插入图片描述
将Ascend-pyACL_5.0.3_linux-aarch64.run传到200DK上。

运行如下命令:

chmod +x Ascend-pyACL-{software_version}.rc1-linux.aarch64.run
./Ascend-pyACL-{software_version}.rc1-linux.aarch64.run --install --run

这个时候终于可以使用ATC工具来完成模型转换了。

运行时找不到opencv:安装opencv-python包(注意:使用pip3.7.5前最好升级一下pip3.7.5到最新,不到最新的话下载opencv会卡住,实际是失败了,只是pip没报错而已)

运行查看结果

运行命令:

python3.7.5 src/classify.py ./data/

最终结果:
在这里插入图片描述

可能的报错:

运行时找不到opencv:安装opencv-python包(注意:使用pip3.7.5前最好升级一下pip3.7.5到最新,不到最新的话下载opencv会卡住,实际是失败了,只是pip没报错而已)

Logo

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

更多推荐