[发明专利]关键点检测方法、装置、电子设备及存储介质

说明书

本发明提供了一种关键点检测方法、装置、电子设备及存储介质，该方法包括：通过主干网络提取待检测图像的图像特征，得到特征图，所述待检测图像包括目标对象；根据多个姿态关键点模板，对所述特征图进行关键点检测，得到所述目标对象对应的至少一组关键点，其中，所述姿态关键点模板表征所述姿态关键点模板中多个关键点的相对位置关系；对所述至少一组关键点进行筛选，得到所述目标对象的关键点检测结果。本发明由于基于多个姿态关键点模板可以直接确定待检测图像中所有目标对象的关键点检测结果，从而可以提高关键点的检测效率。

技术领域

本发明涉及图像识别技术领域，特别是涉及一种关键点检测方法、装置、电子设备及存储介质。

背景技术

关键点检测在日常生活中的应用十分广泛，常见的人脸识别算法通常也依赖于关键点的检测，试妆、美颜、换脸等风靡一时的应用背后也是基于关键点检测技术，这也从另一方面强调了关键点检测精度的高要求。

现有技术中，常见的关键点检测方法通常是两阶段的，第一阶段由一个目标检测模型得到图像中目标对象的位置，第二阶段根据检测出来的目标框将目标对象抠出来再由关键点检测模型进行关键点的检测，这种方式通常也被称为“自顶向下”的方式。这种方式由于需要分步进行，在一个图像中包括多个目标对象时，需要进行多次抠图，而且需要多次使用关键点检测模型分别进行检测，检测效率较低。

发明内容

鉴于上述问题，提出了本发明实施例以便提供一种克服上述问题或者至少部分地解决上述问题的一种关键点检测方法、装置、电子设备及存储介质。

依据本发明实施例的第一方面，提供了一种关键点检测方法，包括：

通过主干网络提取待检测图像的图像特征，得到特征图，所述待检测图像包括目标对象；

根据多个姿态关键点模板，对所述特征图进行关键点检测，得到所述目标对象对应的至少一组关键点，其中，所述姿态关键点模板表征所述姿态关键点模板中多个关键点的相对位置关系；

对所述至少一组关键点进行筛选，得到所述目标对象的关键点检测结果。

依据本发明实施例的第二方面，提供了一种关键点检测方法，包括：

通过主干网络提取待检测图像的图像特征，得到特征图，所述待检测图像包括目标对象；

根据多个姿态关键点模板，对所述特征图进行关键点检测，得到所述目标对象对应的至少一组关键点偏移量，其中，所述姿态关键点模板表征所述姿态关键点模板中多个关键点的相对位置关系，每组关键点偏移量表征所述特征图中该组关键点与每个姿态关键点模板中关键点的偏移量；

根据所述至少一组关键点偏移量，得到所述目标对象的关键点检测结果。

依据本发明实施例的第三方面，提供了一种关键点检测装置，包括：

特征提取模块，用于通过主干网络提取待检测图像的图像特征，得到特征图，所述待检测图像包括目标对象；

关键点检测模块，用于根据多个姿态关键点模板，对所述特征图进行关键点检测，得到所述目标对象对应的至少一组关键点，其中，所述姿态关键点模板表征所述姿态关键点模板中多个关键点的相对位置关系；

检测结果确定模块，用于对所述至少一组关键点进行筛选，得到所述目标对象的关键点检测结果。

依据本发明实施例的第四方面，提供了一种关键点检测装置，包括：

特征提取模块，用于通过主干网络提取待检测图像的图像特征，得到特征图，所述待检测图像包括目标对象；