当前位置:   article > 正文

NCNN人脸检测和关键点检测_onxx 人脸识别模型

onxx 人脸识别模型

参考以下连接获取原始 retinaface模型:

mirrors / wzj5133329 / retinaface_caffe · GitCode

https://github.com/deepinsight/insightface/tree/master/RetinaFace

retinaface(mxnet)的CAFFE模型

https://github.com/Charrin/RetinaFace-Cpp/tree/master/convert_models/mnet

使用caffe2ncnn将模型转换为NCNN格式,需要注意的是,环境需要事先安装protobuf的开发环境,否则CMAKE阶段会警告无法生成caffe2ncnn和onnx2ncnn的问题,因为这两种格式用的都是protobuf来描述的。

  1. sudo apt-get install libprotobuf-dev protobuf-compiler
  2. sudo apt-get autoremove --purge protobuf-compiler
  3. sudo apt-get autoremove --purge libprotobuf-dev

编译成功后,执行命令./caffe2ncnn mnet.prototxt mnet.caffemodel进行模型转换即可。

之后重命名ncnn.bin,ncnn.param为应用识别的文件名,即可以推理验证:

执行命令:./retinaface ./beauty.jpg

lena大妈

百度人脸的结果: 

对结果进行检测:

以上是用NCNN自带的模型测试的效果,下面用自己转的模型验证:

按照符合网络输入要求的图像尺寸做scale后输入:

效果很好!

下面是用NCNN YOLOV5用的模型文件,对比一下人脸检测和人形目标检测算法的区别:

小鲜肉和老腊肉:

网络分析:

一输入九输出

基于NCNN修改的后处理程序:

  1. // Tencent is pleased to support the open source community by making ncnn available.
  2. //
  3. // Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
  4. //
  5. // Licensed
声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/凡人多烦事01/article/detail/284971
推荐阅读
相关标签
  

闽ICP备14008679号