推理模型应用的架构如下
安装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没报错而已)