[发明专利]骨架检测方法、装置、电子设备和计算机可读存储介质

说明书

本发明实施例提出一种骨架检测方法、装置、电子设备和计算机可读存储介质，涉及计算机视觉技术领域。该方法通过将待检测图像输入训练好的骨架检测模型，依次经过特征提取网络、特征融合网络、矢量偏移网络和输出卷积层的处理，即可得到该待检测图像的骨架检测结果。由于训练骨架检测模型所需的骨架标签是通过对原始骨架标签进行膨胀来获得的，故有效改善了监督过程中的任务难度，提升了网络对骨架周围的检测能力；并且，在骨架检测模型的网络结构中引入矢量偏移网络，对检测的骨架信息进行偏移，实现输出的骨架信息的收缩。如此，通过骨架标签的膨胀和输出的骨架信息的收缩，极大提升了整个模型的骨架检测精度。

技术领域

本发明涉及计算机视觉技术领域，具体而言，涉及一种骨架检测方法、装置、电子设备和计算机可读存储介质。

背景技术

骨架是Blum提出的一种图像特征，骨架是物体的一维简化表示方式，骨架同时包含物体的结构信息和物体部件之间的连接信息，可以准确表达物体的几何特征和拓扑特征。物体的骨架也称作中轴，是一种基于物体结构的描述符。因为骨架像素仅仅占据了图像中像素很少的比例，因此骨架具有相当大的抽象性。骨架通常是从对象中抽取出来的结构信息，它对非刚性形变具有不变性，对物体骨架建模在计算机视觉中具有广泛的应用。

从自然图像中直接检测物体骨架是一个热点问题，同时也面临很多挑战。自然图像中的物体往往呈现多种形态、颜色、纹理以及形状，另外，自然场景下获得的图像难免受到光照不均匀、对比度低、遮挡等问题的影响。因此，能否准确提取出骨架也将在一定程度上影响后续其他智能系统的处理性能。

已有的骨架检测技术，主要通过使用多个骨架滤波器，每个骨架滤波器均具有零和性结构及反射对称性结构，实现对不具有对称结构的噪声分布进行过滤，而有效的骨架被保留。这种通过使用多个骨架滤波器对高噪声图像进行无监督的骨架提取的方式，在面对复杂场景时，物体骨架检测精度会降低。

发明内容

有鉴于此，本发明的目的在于提供一种骨架检测方法、装置、电子设备和计算机可读存储介质，以解决现有的骨架检测技术存在的物体骨架检测精度低的问题。

为了实现上述目的，本发明实施例采用的技术方案如下：

第一方面，本发明提供一种骨架检测方法，所述方法包括：

将待检测图像输入训练好的骨架检测模型；其中，训练所述骨架检测模型所采用的骨架标签是对原始骨架标签进行膨胀后获得；

利用所述骨架检测模型的特征提取网络对所述待检测图像进行特征提取，得到多个目标特征图；

将所述多个目标特征图输入所述骨架检测模型的特征融合网络，利用所述特征融合网络对所述多个目标特征图进行特征融合，得到融合后的特征图；

将所述融合后的特征图输入所述骨架检测模型的矢量偏移网络，利用所述矢量偏移网络对所述融合后的特征图进行偏移处理，得到偏移后的特征图；

将所述偏移后的特征图输入所述骨架检测模型的输出卷积层，利用所述输出卷积层对所述偏移后的特征图进行卷积操作，得到所述待检测图像对应的骨架检测结果。

在可选的实施方式中，所述骨架标签通过以下方式生成：

依据预设的半径阈值对原始骨架标签进行膨胀操作，得到膨胀后的骨架标签；

计算所述膨胀后的骨架标签中每个非零像素点到最近零值点的距离信息；

根据所述距离信息和预设的高斯分布函数，得到高斯分布膨胀骨架；

根据所述高斯分布膨胀骨架中的最大值对所述高斯分布膨胀骨架进行归一化处理，得到所述骨架标签。