手把手教你 在Pytorch框架上部署和测试关键点人脸检测项目DBFace
liuian 2025-07-24 19:47 7 浏览
这期教向大家介绍仅仅 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、深度学习和计算机编程的精彩内容,可以关注微信公众号:码农的后花园
相关推荐
- 基于Spring Security的JWT认证深度解析与实践指南
-
一、JWT认证核心原理剖析1.JWT令牌结构解析Header:采用HMACSHA256算法示例json{"alg":"HS256","typ&...
- 入门到精通:电商API的全栈开发指南
-
在当今电商蓬勃发展的时代,API(应用程序编程接口)作为系统间的“桥梁”,已成为构建高效、可扩展电商平台的核心。全栈开发涉及从前端用户界面到后端服务器、数据库的完整流程,本指南将带你从零基础逐步掌握电...
- SpringBoot整合SpringSecurity+JWT
-
作者|Sans_https://juejin.im/post/5da82f066fb9a04e2a73daec一.说明SpringSecurity是一个用于Java企业级应用程序的安全框架,主要包含...
- Spring Boot 集成 JWT 详细指南(springboot集成junit)
-
SpringBoot集成JWT详细指南在当今的Web应用开发中,安全认证是至关重要的一环。JSONWebToken(JWT)作为一种轻量级的认证和授权机制,因其跨域支持、自包含性以及易...
- 电商API安全最佳实践:保护用户数据免受攻击
-
在电商领域,API(应用程序编程接口)是连接用户、商家和支付系统的核心枢纽。它们处理敏感数据,如用户个人信息、支付详情和交易记录。然而,API也常成为黑客攻击的目标,导致数据泄露、欺诈和声誉损失。本文...
- Egg.js 异常处理、中间件、jwt,实现接口权限控制
-
一、自定义异常、异常处理中间件在程序执行时会有各种各样的异常情况,当异常出现我们能从控制台看出异常的原因,但是对前端来说不够人性化,不能够清晰,有些情况要给调用端返回友好的消息提示,利用自定义异常和全...
- list列表基本操作(list list)
-
【实验目的】1、掌握list列表的基本操作【实验原理】列表是Python中最基本的数据结构,列表是最常用的Python数据类型,列表的数据项不需要具有相同的类型。列表中的每个元素都分配一个数字-它...
- 基于Python的多人拼图游戏(python项目拼图游戏)
-
开发环境要求本系统的软件开发及运行环境具体如下。操作系统:Windows7、Windows10。Python版本:Python3.7.0。开发工具:PyCharm。运行方法在项目文件中找到mai...
- python开发工具PyCharm最新版本新增功能介绍
-
PyCharmV2022.1支持自定义包存储库的身份验证、改进的TypedDict和Docker的新服务UI。PyCharm官方最新版免费下载试用,历史版本下载,在线文档和帮助文件下载-慧都网...
- 走马观花看PySide6(官方examples)
-
准备工作1、安装PythonPython下载地址:https://www.python.org/downloads/2、安装PyCharmPyCharm下载地址:https://www.jetbrai...
- 【0基础学爬虫】爬虫基础之scrapy的使用
-
【0基础学爬虫】爬虫基础之scrapy的使用大数据时代,各行各业对数据采集的需求日益增多,网络爬虫的运用也更为广泛,越来越多的人开始学习网络爬虫这项技术,K哥爬虫此前已经推出不少爬虫进阶、逆向相关文章...
- 手把手教你 在Pytorch框架上部署和测试关键点人脸检测项目DBFace
-
这期教向大家介绍仅仅1.3M的轻量级高精度的关键点人脸检测模型DBFace,并手把手教你如何在自己的电脑端进行部署和测试运行,运行时bug解决。01.前言前段时间DBFace人脸检测库横空出世,...
- Scrapy框架的安装(scrapy框架图)
-
Win+R输入cmd打开命令行我们先把pip升级到最新版,输入代码如下:pipinstall--upgradepip不过一般这种更新方式会经常性出错,安装文件在下载到一半时就会超时报错可以试试...
- Pycharm设置本地Python项目解释器
-
设置Pycharm项目运行的python环境,由于不同的项目使用的python环境是不一样的,需要根据项目配置不同的python环境,比如在电脑上面安装了Python3.11.x和3.9.x两个环境,...
- Python支付宝单笔转账接口(python支付宝自动转账)
-
开发信息接口加签方式为证书模式证书模式好处是可以使用支付宝的转账到支付宝账户,也就是提现功能,公钥模式不能实现转账到支付宝账户。此DEMO利用单笔转账到支付宝账户接口【提现功能】用户可以通过此DEMO...
- 一周热门
-
-
Python实现人事自动打卡,再也不会被批评
-
【验证码逆向专栏】vaptcha 手势验证码逆向分析
-
Psutil + Flask + Pyecharts + Bootstrap 开发动态可视化系统监控
-
一个解决支持HTML/CSS/JS网页转PDF(高质量)的终极解决方案
-
再见Swagger UI 国人开源了一款超好用的 API 文档生成框架,真香
-
网页转成pdf文件的经验分享 网页转成pdf文件的经验分享怎么弄
-
C++ std::vector 简介
-
系统C盘清理:微信PC端文件清理,扩大C盘可用空间步骤
-
10款高性能NAS丨双十一必看,轻松搞定虚拟机、Docker、软路由
-
飞牛OS入门安装遇到问题,如何解决?
-
- 最近发表
- 标签列表
-
- python判断字典是否为空 (50)
- crontab每周一执行 (48)
- aes和des区别 (43)
- bash脚本和shell脚本的区别 (35)
- canvas库 (33)
- dataframe筛选满足条件的行 (35)
- gitlab日志 (33)
- lua xpcall (36)
- blob转json (33)
- python判断是否在列表中 (34)
- python html转pdf (36)
- 安装指定版本npm (37)
- idea搜索jar包内容 (33)
- css鼠标悬停出现隐藏的文字 (34)
- linux nacos启动命令 (33)
- gitlab 日志 (36)
- adb pull (37)
- table.render (33)
- python判断元素在不在列表里 (34)
- python 字典删除元素 (34)
- vscode切换git分支 (35)
- python bytes转16进制 (35)
- grep前后几行 (34)
- hashmap转list (35)
- c++ 字符串查找 (35)