使用MTCNN进行人脸检测

本文介绍了使用MTCNN进行人脸检测的详细过程,包括图像金字塔、PNet、RNet和ONet的工作原理及实现细节。通过参考特定代码库,解释了网络结构和输出,并讨论了校准和非极大值抑制在算法中的作用。

人脸检测是人脸识别中重要的一环,MTCNN是比较常用的人脸检测算法,这里主要介绍测试过程。
论文地址

主要流程:

在这里插入图片描述
1.做图片金字塔,即将图片resize成不同大小
2.将不同大小的图片输入PNet,得到候选框的偏移值,映射到原图得到许多候选框,做非极大值抑制(nms)去掉部分候选框
3.去原图截取候选框的图像(要截取覆盖这个候选框的最大正方形框),resize成24x24,输入RNet,得到候选框偏移值和分数,继续nms筛选
4.同样方法截取剩余的候选框,resize成48x48,输入ONet,经过nms筛选后得到最终人脸框,同时输出人脸特征点位置

实现细节:

参考代码:https://github.com/TreB1eN/InsightFace_Pytorch 中的mtcnn-pytorch版本,里面有手把手教你怎么做。
不过网络有个bug,PNet中forward函数里“a = F.softmax(a, dim=-1)”,dim应该改为1。也不知道为啥他们运行都没问题。。。

首先关于PNet的输入输出,网络结构如图:
在这里插入图片描述
输入:图中输入的宽和高为12X12,这是最小的输入单元,也就是检测到的最小人脸大小。如果要检测大于12X12的

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值