人脸关键点检测模型的训练方法和人脸关键点检测方法与流程

文档序号:26174928发布日期:2021-08-06 18:20阅读:来源:国知局

技术特征:

1.一种人脸关键点检测模型的训练方法,其特征在于,包括:

获取携带有预置人脸标注信息的人脸样本图像;

将所述人脸样本图像输入至mtcnn网络的pnet网络中,得到所述人脸样本图像对应的人脸候选框;

将所述人脸候选框、所述人脸样本图像输入至所述mtcnn网络的rnet网络中,得到所述人脸样本图像对应的人脸目标框;

将所述人脸目标框、所述预置人脸标注信息和所述人脸样本图像输入至所述mtcnn网络的onet网络中,得到训练好的mtcnn网络模型,其中,所述onet网络中的关键点回归函数为:预置wingloss函数。

2.根据权利要求1所述的训练方法,其特征在于,所述预置wingloss函数为:

其中,loss(x)为预置wingloss函数,w为非线性部分之间的间隔,ε为曲率,用于限制曲线的弯曲程度,c为常量,x为预测值与真实值之间的差值。

3.根据权利要求1所述的训练方法,其特征在于,所述人脸标注信息包括:人脸标注框和人脸关键点;

所述获取携带有预置人脸标注信息的人脸样本图像,具体包括:

获取未经标注的原始人脸样本图像;

通过预置标注工具,对所述原始人脸样本图像进行人脸框标注,得到标注有人脸标注框的人脸样本图像;

通过预置检测接口,对所述原始人脸样本图像进行关键点检测,得到关键点坐标;

根据所述关键点坐标,对所述原始人脸样本图像进行关键点标注,得到标注有人脸关键点的人脸样本图像。

4.根据权利要求3所述的训练方法,其特征在于,所述人脸候选框包括:正人脸候选框、中人脸候选框和负人脸候选框;

所述将所述人脸样本图像输入至mtcnn网络的pnet网络中,得到所述人脸样本图像对应的人脸候选框,具体包括:

将所述人脸样本图像输入至mtcnn网络的pnet网络,使得所述pnet网络对所述人脸样本图像进行人脸框选取,并基于各人脸选取框对应的交叠率,将所述选取框分类为正人脸候选框、中人脸候选框和负人脸候选框。

5.根据权利要求4所述的训练方法,其特征在于,所述对所述人脸样本图像进行人脸框选取,并基于各人脸选取框对应的交叠率,将所述选取框分类为正人脸候选框、中人脸候选框和负人脸候选框,具体包括:

对所述人脸样本图像进行人脸框选取,得到若干人脸选取框;

计算每一所述人脸选取框和所述人脸标注框之间的交叠率;

根据各所述人脸选取框对应的所述交叠率,对所述人脸选取框进行分类,得到正人脸候选框、中人脸候选框和负人脸候选框。

6.根据权利要求5所述的训练方法,其特征在于,所述根据各所述人脸选取框对应的所述交叠率,对所述人脸选取框进行分类,得到正人脸候选框、中人脸候选框和负人脸候选框,具体包括:

将所述交叠率大于等于第一阈值的所述人脸选取框作为正人脸候选框;

将所述交叠率大于第二阈值小于所述第一阈值的所述人脸选取框作为所述中人脸候选框;

将所述交叠率小于等于所述第二阈值的所述人脸选取框作为负人脸候选框,其中,所述第二阈值小于所述第一阈值。

7.一种人脸关键点检测方法,其特征在于,包括:

获取待检测人脸图像;

将所述待检测人脸图像输入至预置mtcnn网络模型中,得到所述预置mtcnn网络模型输出的所述待检测人脸图像的关键点信息,其中,所述预置mtcnn网络模型为根据权利要求1至6中任一项所述的训练方法训练得到。

8.一种人脸关键点检测模型的训练装置,其特征在于,包括:

获取单元,用于获取携带有预置人脸标注信息的人脸样本图像;

第一处理单元,用于将所述人脸样本图像输入至mtcnn网络的pnet网络中,得到所述人脸样本图像对应的人脸候选框;

第二处理单元,用于将所述人脸候选框、所述人脸样本图像输入至所述mtcnn网络的rnet网络中,得到所述人脸样本图像对应的人脸目标框;

第三处理单元,用于将所述人脸目标框、所述预置人脸标注信息和所述人脸样本图像输入至所述mtcnn网络的onet网络中,得到训练好的mtcnn网络模型,其中,所述onet网络中的关键点回归函数为:预置wingloss函数。

9.根据权利要求8所述的训练装置,其特征在于,所述预置wingloss函数为:

其中,loss(x)为预置wingloss函数,w为非线性部分之间的间隔,ε为曲率,用于限制曲线的弯曲程度,c为常量,x为预测值与真实值之间的差值。

10.一种人脸关键点检测装置,其特征在于,包括:

获取单元,用于获取待检测人脸图像;

检测单元,用于将所述待检测人脸图像输入至预置mtcnn网络模型中,得到所述预置mtcnn网络模型输出的所述待检测人脸图像的关键点信息,其中,所述预置mtcnn网络模型为根据权利要求1至6中任一项所述的训练方法训练得到。


技术总结
本申请公开了一种人脸关键点检测模型的训练方法和人脸关键点检测方法,其中训练方法包括:获取携带有预置人脸标注信息的人脸样本图像;将所述人脸样本图像输入至MTCNN网络的Pnet网络中,得到所述人脸样本图像对应的人脸候选框;将所述人脸候选框、所述人脸样本图像输入至所述MTCNN网络的Rnet网络中,得到所述人脸样本图像对应的人脸目标框;将所述人脸目标框、所述预置人脸标注信息和所述人脸样本图像输入至所述MTCNN网络的Onet网络中,得到训练好的MTCNN网络模型,其中,所述Onet网络中的关键点回归函数为:预置WingLoss函数,解决了现有的人脸关键点检测方法在进行人脸检测时,容易出现关键点漂移,导致人脸关键点检测准确度较低的技术问题。

技术研发人员:刘思伟
受保护的技术使用者:广州织点智能科技有限公司
技术研发日:2021.05.26
技术公布日:2021.08.06
当前第2页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1