张小白带你玩转Atlas 200DK+CANN 5.0.4+MindStudio 3.0.4(16)尝试昇思跑LeNet-4
十六、在Atlas 200DK上运行MindSpore LeNet网络的推理代码
书接上回:https://bbs.huaweicloud.com/blogs/349622
经过多次试验,张小白将run_infer_310.sh文件做如下修改:
(1)将python命令均改为python3,因为当前环境python指向的是Python2。
(2)在执行推理命令之前,新增 以下语句:
export LD_LIBRARY_PATH=HOME/Ascend/ascend−toolkit/5.0.4.alpha005/arm64−linux/opp/opimpl/built−in/aicore/tbe/optiling:LD_LIBRARY_PATH
. /home/HwHiAiUser/Ascend/ascend-toolkit/set_env.sh
(3)将执行语句最后的 &>xx.log &去掉,将错误信息打印在屏幕上:
(4)修改main.cc文件,增加打印文件名的代码:
执行run_infer_310.sh命令并查看结果:
bash ./run_infer_310.sh /home/HwHiAiUser/models/official/cv/lenet/lenet.mindir /home/HwHiAiUser/models/official/cv/lenet/MNIST/test dvpp 0
。。。
虽然看到在preprocess_Result文件夹下有10000个test jpg文件。
下载下来看了下,都是手写的图片:
但是第一个文件就推理失败了!
只好又提了issue,赵老师回答了我的问题:
如法炮制:
先到X86上,编辑export.py文件:
执行export (命令可参看 https://bbs.huaweicloud.com/blogs/349621 )
python3 export.py --ckpt_file ckpt/checkpoint_lenet-10_1875.ckpt --file_format "MINDIR" --device_target="CPU"
可以看到新的mindir文件也已经生成:
将lenet.mindir拷贝到200DK上。
scp lenet.mindir HwHiAiUser@192.168.199.204:/home/HwHiAiUser/models/official/cv/lenet/
这个是需要注意的,如果200DK上有同名文件,会报权限错误。
那就删除掉200DK的同名文件:
再回到X86开发环境重新scp:
scp lenet.mindir HwHiAiUser@192.168.199.204:/home/HwHiAiUser/models/official/cv/lenet/
再重新执行推理:
bash ./run_infer_310.sh /home/HwHiAiUser/models/official/cv/lenet/lenet.mindir /home/HwHiAiUser/models/official/cv/lenet/MNIST/test dvpp 0
结果如下:
它会完成10000个测试图片的推理,精度结果是 0.9854.
推理结果放到了 result_Files目录下:
可见已经推理成功!
(未完待续)
- 点赞
- 收藏
- 关注作者
评论(0)