前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >基于PaddlePaddle实现的快速人脸识别模型

基于PaddlePaddle实现的快速人脸识别模型

作者头像
夜雨飘零
发布2021-12-07 14:49:47
5850
发布2021-12-07 14:49:47
举报
文章被收录于专栏:CSDN博客CSDN博客

前言

本项目参考了ArcFace的损失函数,同时参考了PP-OCRv2模型结构,意在开发一个模型较小,但识别准确率较高且推理速度快的一种人脸识别项目,该项目训练数据使用emore数据集,一共有85742个人,共5822653张图片,使用lfw-align-128数据集作为测试数据。

源码地址:https://github.com/yeyupiaoling/PaddlePaddle-MobileFaceNets

数据集准备

本项目提供了标注文件,存放在dataset目录下,解压即可。另外需要下载下面这两个数据集,下载完解压到dataset目录下。

然后执行下面命令,将提取人脸图片到dataset/images,并把整个数据集打包为二进制文件,这样可以大幅度的提高训练时数据的读取速度。

代码语言:javascript
复制
python create_dataset.py

训练

执行train.py即可,更多训练参数请查看代码。

代码语言:javascript
复制
python train.py

评估

执行eval.py即可,更多训练参数请查看代码。

代码语言:javascript
复制
python eval.py

预测

本项目已经不教提供了模预测,模型文件可以直接用于预测。在执行预测之前,先要在face_db目录下存放人脸图片,每张图片只包含一个人脸,并以该人脸的名称命名,这建立一个人脸库。之后的识别都会跟这些图片对比,找出匹配成功的人脸。。这里使用的人脸检测是MTCNN模型,这个模型具有速度快,模型小的特点,源码地址:PaddlePaddle-MTCNN

如果是通过图片路径预测的,请执行下面命令。

代码语言:javascript
复制
python infer.py --image_path=temp/test.jpg

日志输出如下:

代码语言:javascript
复制
人脸检测时间:45ms
人脸识别时间:6ms
人脸对比结果: [('杨幂', 0.61594474), ('迪丽热巴', 0.37707973)]
人脸对比结果: [('迪丽热巴', 0.7290128), ('杨幂', 0.3993025)]
预测的人脸位置: [[156, 80, 214, 135, 1], [269, 67, 327, 121, 1]]
识别的人脸名称: ['杨幂', '迪丽热巴']
总识别时间:53ms
在这里插入图片描述
在这里插入图片描述

如果是通过相机预测的,请执行下面命令。

代码语言:javascript
复制
python infer_camera.py --camera_id=0
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2021-11-03 ,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 前言
  • 数据集准备
  • 训练
  • 评估
  • 预测
相关产品与服务
人脸识别
腾讯云神图·人脸识别(Face Recognition)基于腾讯优图强大的面部分析技术,提供包括人脸检测与分析、比对、搜索、验证、五官定位、活体检测等多种功能,为开发者和企业提供高性能高可用的人脸识别服务。 可应用于在线娱乐、在线身份认证等多种应用场景,充分满足各行业客户的人脸属性识别及用户身份确认等需求。
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档