在小藤上实现Retinaface人脸识别
【摘要】 在小藤上实现Retinaface人脸识别
浏览器下载 https://mindx.sdk.obs.cn-north-4.myhuaweicloud.com/mindxsdk-referenceapps%20/contrib/Retinaface/ATC%20Retinaface%28FP16%29%20from%20Pytorch.zip
获得：ATC Retinaface(FP16) from Pytorch.zip
解压
发现了很多文件，有pth模型，onnx模型，有om模型，还有很多脚本。
我们不要紧张。只取onnx文件传到A2的 ~/ascend_community_projects/Retinaface/model目录下
cd ~/ascend_community_projects/Retinaface
cd model
执行模型转换，将onnx模型转为om模型：
sh ./run.sh
模型文件 newRetinaface.om 已生成：
跟推理代码main.cpp中的模型文件是一致的：
下面开始编译推理代码：
修改CMakeList.txt，修改MX_SDK_HOME为 /home/HwHiAiUser/mxVision-5.0.RC2
设置环境变量：
source /usr/local/Ascend/ascend-toolkit/set_env.sh
source /home/HwHiAiUser/mxVision-5.0.RC2/set_env.sh
执行编译：
bash build.sh
准备一张图片：test.jpg
传到A2上:
执行推理：
./retinaface ./test.jpg
结果文件result.jpg也生成了，下载到本地查看：
可见所有MM的脸和五官都被框出来了。
（全文完，谢谢阅读）
【版权声明】本文为华为云社区用户原创内容，转载时必须标注文章的来源（华为云社区）、文章链接、文章作者等基本信息， 否则作者和本社区有权追究责任。如果您发现本社区中有涉嫌抄袭的内容，欢迎发送邮件进行举报，并提供相关证据，一经查实，本社区将立刻删除涉嫌侵权内容，举报邮箱：
cloudbbs@huaweicloud.com
- 点赞
- 收藏
- 关注作者
评论（0）