Skip to content

nniefacelib是一个在海思35xx系列芯片上运行的人脸算法库

License

Notifications You must be signed in to change notification settings

dsx-ai/nniefacelib

 
 

Repository files navigation

nniefacelib

nniefacelib是一个在海思35xx系列芯片上运行的人脸算法库,目前集成了mobilefacenet和retinaface。 后期也会融合一些其他经典的模型,目的也是总结经验,让更多人早日脱离苦海。

目前只在3516DV300上进行了测试。鉴于很多同学在其他型号上进行测试失败的情况,做一下统一说明,编译错误显示, 不同型号的sample代码略有不同,有必要的话需要自己移植核心源码,仿照着SDK的源码进行移植,应该问题不大

更新项

2020-5-15 添加linux mapper工具,以retinaface为例子
https://pan.baidu.com/s/16pFjdX_fQcVl2I4bAIf4Mw
code:1dw8

2020-4-28 添加了retinaface与mobilefacenet的merge bn tool, 提供了优化后的低精度retinaface(640x640), 17ms, 而高精度为34ms

2020-4-24 更新了PFPLD模型在海思上的前向传播代码

入坑指南

因为代码这部分并没有对NNIE进行重构,是纯C代码,所以看起来会比较晦涩,为此,我们可以将算法模块直接编译成 动态链接库供上层进行调用。

make -f Makefile.Shared

为了方便测试,也可以直接编译成可执行文件

make -f Makefile.Debug
# 人脸检测
./nnie_face_test 0
# 人脸识别特征提取
./nnie_face_test 1
# PFPLD人脸姿态估计以及98关键点
./nnie_face_test 2

模型转换

模型转换其实是一个大坑,也可以说是一个玄学,如何转换?如何让速度更快?目前也是在总结一些经验, 我也会在知乎上更新一些关于模型转换的一些方法
海思NNIE之Mobilefacenet量化部署
海思NNIE之RetinaFace量化部署
海思NNIE之PFPLD训练与量化

TODO

About

nniefacelib是一个在海思35xx系列芯片上运行的人脸算法库

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • C 47.5%
  • Python 26.3%
  • C++ 25.8%
  • Other 0.4%