本文介绍了将libfacedetection的人脸检测模型移植到海思hi3559a板子上的过程，包括模型转换、后处理实现以及推理时间的优化。在nnie上推理时间为10ms，后处理时间5-10ms。通过调整模型，将softmax移入nnie处理，但推理时间增加到18ms，未来计划进一步优化并开源源代码。
人脸检测模型：libfacedetection
模型开源项目地址：https://github.com/ShiqiYu/libfacedetection （感谢于老师）
嵌入式平台：hi3559a
nnie推理时间：10ms
1 转模型
在项目中可以找到caffe模型，笔者使用的是：yufacedetectnet-open-v2.prototxt和yufacedetectnet-open-v2.caffemodel。转模型前需要对prototxt文件进行修改，因为有些层nnie是不支持的。修改如下：
（1）input_shape改为1*3*576*1024
（2）去掉了loc和conf的flatten、permute、concat、detection_out、PriorBox等网络层，在c代码中的后处理中实现。下面是笔者修改过的prototxt文件
yufacedetectnet-open-v2.prototxt
18.2K
·
百度网盘
（3）使用nnie_mapper_11工具转模型，mapper工具需要读取.cfg文件，用于配置转模型的参数，具体的参数详解可以参考海思的svp文档，下面是笔者使用的.cfg文件
最低0.47元/天 解锁文章
556
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
