百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 技术文章 > 正文

手把手教你 在Pytorch框架上部署和测试关键点人脸检测项目DBFace

itomcoil 2025-06-18 21:43 5 浏览

这期教向大家介绍仅仅 1.3M 的轻量级高精度关键点人脸检测模型DBFace,并手把手教你如何在自己的电脑端进行部署和测试运行,运行时bug解决。

01. 前言

前段时间DBFace人脸检测库横空出世,但是当时这个人脸识别模型是7M大小,几乎可以识别出世界最大自拍中的所有人像。DBFace出自国内人工智能公司深兰科技(DeepBlue),这个模型的创建者正是这个公司的两位“高手”-Libia和Wish,而最近,高手就是高手,两位大佬对模型进行了再升级,现在这个模型的大小仅仅只有1.3M。

该模型已经放在GitHub上开源了,被《机器之心》挖掘出来,目前在Github上的Stars数已经有800颗星了,于是带着好奇和感兴趣我想要试试这个人脸检测模型,没想到,测试效果极好,这里就和大家分享一下。

02. DBFace的网络结构

DBFace的初衷设计就是为了设计成一款轻量级的人脸检测器,能够在边缘计算上有效的使用,为了让处理更加简洁、高效、对小目标的检测效果也好,选择 CenterNet(基于中心点的方法预测目标,称为:CenterNet,)结构做检测任务,采用MoblienetV3做Backbone。我相信大家在这里已经晕了吧,不知道什么是CenterNet和MoblienetV3,因为这不是几句话能说清楚的,需要大家明白卷积的知识,在知道这之后,还需要去读这个论文,啃代码,可能才能理解个七八分吧。这里给大家两个链接:

CenterNet:https://blog.csdn.net/c20081052/article/details/89358658

MobileNetV3:https://blog.csdn.net/DL_wly/article/details/90168883

感兴趣的话就继续啃吧

相对于MoblienetV2,MoblienetV3在其基础上新增了SE、Hard-Swish Activation等模块,在兼顾infence速度的同事提升网络性能。对于任务头的设计,大佬引入了SSH的DetectModule和ContextMoudle,经试验验证该模块能有效的提升DBFace算法的检出性能,这里涉及到卷积、池化中的最大值池,对模型进行的整合。


对于上采样设计,作者采用Upsample + Conv + Act的方式,即上采样 + 卷积 + 激活函数Activation,从图中可以看到有激活函数Relu。

03. 损失函数Loss的定义

整个网络由三个部分的Loss组成:

  • 热力图HeatMap损失 - (CenterNet网络中热力图误差损失)
  • 位置坐标偏移量(Bounding Box)损失 - (检测边界框误差损失)
  • 关键点(Landmark)损失 - (人脸关键点误差损失)

对于一个模型来说,网络结构和损失函数、以及数据集是最重要的,这里作者并没有开源自己的数据集,我们可以利用工具来标记属于自己的数据集,这里给大家介绍一个关键点标记的软件CasiaLabeler,可以在后台回复关键字"关键点"获取。

04. 项目运行效果

05. 官方运行效果图:


06. 项目部署和测试

第一步:Github上下载项目源码,选择Clone and Download ,ZIP格式下载到桌面,解压缩,地址:https://github.com/dlunion/DBFace,也可以关注微信公众号:码农的后花园,直接获取。

第二步:工具准备

Anaconda + Pycharm + CUDA10,还未下载安装的同学,可以看我之前的下载安装详细教程。

第三步:在Anaconda中创建一个python3.6及以上的虚拟环境,环境名为py36,否则会出现在字符串f ,SyntaxError错误,这种用法是python3.6之后才用的,不会创建的看之前的教程。

第四步:在虚拟环境中下载项目依赖包

Pytorch1.5 + OpenCV-Python 4.2,到官网进行下载安装,还不会的可以看我之前写的教程。

第五步:使用Pycharm打开已经解压好的项目,然后再pycharm中配置好python解释器,选择刚刚创建好了的py36环境,点击Add->找到Anaconda的安装目录下的env目录下你已经创建好的环境名为py36.

第六步:找到一张图片,将其放在项目的datas文件目录下,给图片命名为party.jpg

第七步:进行测试运行main.py,运行之前,在image_demo中添加你想要测试的图片代码


第八步:右键main.py,选择Run 'main',运行完之后在detect_result里查看检测效果,也可以调用PC端的摄像头,进行实时人脸检测,选择camera_demo即可。

关注微信公众号,后台回复关键字:人脸识别,即可获取人脸数据集标记工具和训练模型。

至此完成成功,大家可以自己动手试一试,觉得还不错的话记得点赞收藏哦,或者分享给身边的好友们,码字不易,希望得到你们的支持。

更多有关python、深度学习和计算机编程的精彩内容,可以关注微信公众号:码农的后花园

相关推荐

基于Python开发的家居用品外贸网站(B2B企业官网)

作为一名程序员,平时除了本职工作,我也会利用业余时间做一些兼职和副业,大部分的私活都是从某鱼和CSDN接的,有些是别人介绍的。最近刚好接到了一个外贸企业网站的开发需求,客户是一家位于深圳的家居用品公司...

摄像头视频流处理方法总结,如何掌握视频流处理技巧?

摄像头视频流处理是指其技术本质、关键处理流程和核心目标。1.视频流的定义视频流(VideoStream)是摄像头连续采集图像帧(frame),并以压缩编码格式(如H.264/H.265)传输或存储...

大学生机器人开发辅导|代码、机构到控制全链路进阶

想做机器人项目,却苦于不会硬件、算法和代码?SRTP、大创、全国竞赛动手项目无从下手?Arduino、STM32、RaspberryPi模块让你头大?想在毕业设计、创新实践、社团活动做出有亮点的机器...

OpenCV实现手势音量控制(opencv按钮)

前言:Hello大家好,我是Dream。今天来学习一下如何使用OpenCV实现手势音量控制,欢迎大家一起前来探讨学习~一、需要的库及功能介绍本次实验需要使用OpenCV和mediapipe库进行手...

DIY激光枪薄纱蟑螂!AI杀蚊子博士新作,项目已开源

萧箫发自凹非寺量子位|公众号QbitAI几个世纪来,人类都处在被蟑螂支配的恐惧中。但比蟑螂更恐怖的,是打不死还消失了的蟑螂……现在,一位博士搞出了一套自动识别并薄(爆)纱(杀)蟑螂的激光“炮...

为了杀蚊子,这位博士用树莓派DIY了一把激光枪

金磊发自凹非寺量子位报道|公众号QbitAI世人苦蚊子久矣。尤其在夏夜,耳边嗡嗡作响,甚至还得与其“挑灯夜战”个三百回合。为此,一个国外博士便DIY了一种高端的杀蚊方式:计算机视觉精准定...

草根PLC革命:低价魔改开源方案对决高价LabVIEW工业视觉系统!

魔改方案:三菱FX5U+树莓派魔改架构硬件清单创新设计:双核协同架构:PLC负责实时控制+树莓派运行AI算法EtherCAT菊花链:省交换机成本,布线效率提升70%五大开源工具:CODESYSR...

机器人开发进阶:看懂这五个项目中的软硬件哲学

DIY机器人项目正逐渐成为技术爱好者和创客们的热门选择。无论是用于教育、娱乐还是实际应用,机器人技术都展现了其强大的潜力。本文将介绍五个EEWorld上备受工程师关注的五个DIY项目,通过本文,您将深...

树莓派到底是什么?能干什么?有必要买吗?

很多人疑惑,树莓派到底是什么?能干什么?有必要买吗?今天我带着这三个疑惑给大家解答;一、树莓派到底是什么?树莓派通俗的理解就是一台便携式小型电脑,起码最新的树莓派4当一台电脑是可以的;往深了说,他可以...

用腾讯优图AI视觉模组做一个驾驶疲劳监测仪

道路千万条,安全第一条,相信每个人都是牢记于心的,“喝酒不开车,开车不喝酒”其实不难,难的是防范始料未及的事件,疲倦就是众多始料未及事件中,杀伤力稳居前排的,前一秒心里还在想坚持坚持,下一秒可能就失去...

Z410升级树莓派4B机型终于和大家见面了

*Z410-4B入门二次开发平台*经过近1年的反复测试,Z410升级树莓派4B的机型终于和大家见面了!Z410机型设计的初衷,就是想为大家提供一款基础的、开源的、高性价比、可扩展、可进行二次开发的无人...

有保险柜怕不安全,用树莓派制作一款只有刷脸才能开的保险柜吧

眨眨眼睛就能保住身家,好过记住密码或拿着钥匙,你的脸就是保险柜的钥匙!这个作品将展示给大家如何用树莓派和摄像头制作一个人脸识别的保险柜,当然,如果不慎毁容或整容,不好意思。。。直接拿锤子砸了吧软件部分...

项目分享|仅需1板卡+1摄像头,3步完成人脸喜怒哀乐识别

使用OpenCV、TensorFlow和Keras,基于RaspberryPi进行情绪识别,你的心情一览无余。面部表情识别系统可用于多种应用,可以用来研究或分析人的情绪。许多公司正在植入...

价值8美元的OpenAsk收费问题回答:孩子小学五年级如何入门编程?

从今天起,给大家持续更新OpenAsk上的一些收费问题的回答系列,内容信不信由你,但是如果您仔细看了分析后会说,真香。#少儿编程是智商税吗#“更多内容欢迎关注-司马南柯一梦”(欢迎随意转发)下面是一位...

再见 Pycharm,这款开箱即用的轻量级神器你值得拥有

作者:豆豆来源:Python技术如果你问我最好用的IDE是什么,那我肯定会毫不犹豫的告诉你Pycharm。毕竟jetbrains出品必属精品。但对于很多初学者来讲,Pycharm显得略笨...