张小白带你玩转Atlas 200DK+CANN 5.0.4+MindStudio 3.0.4(16)尝试昇思跑LeNet-4

举报
张辉 发表于 2022/04/26 21:12:10 2022/04/26
【摘要】 Atlas 200DK+CANN 5.0.4+MindStudio 3.0.4+MindSpore 1.6.1

十六、在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目录下:

可见已经推理成功!

(未完待续)

【版权声明】本文为华为云社区用户原创内容,转载时必须标注文章的来源(华为云社区)、文章链接、文章作者等基本信息, 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@huaweicloud.com
  • 点赞
  • 收藏
  • 关注作者

评论(0

0/1000
抱歉,系统识别当前为高风险访问,暂不支持该操作

全部回复

上滑加载中

设置昵称

在此一键设置昵称,即可参与社区互动!

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。

*长度不超过10个汉字或20个英文字符,设置后3个月内不可修改。