[发明专利]一种金属裂纹检测方法、装置、电子设备及存储介质

说明书

本公开涉及图像检测技术领域，提供了一种金属裂纹检测方法、装置、电子设备及存储介质。该方法包括：将待检图片输入预设的特征提取网络，经特征提取网络的卷积层处理后，输出特征图；使用特征提取网络的检测层对特征图进行再次特征提取和分析，得到金属裂纹检测结果，金属裂纹检测结果包括第一、第二、第三和第四特征图；其中，第一特征图用于表征是否存在目标物体，第二特征图用于表征目标物体的中心点相较于第一特征图的特征点的偏移量，第三特征图用于表征目标物体对应的目标框的宽高，第四特征图用于表征目标框的倾斜角度。本公开能够很好地提高金属裂纹的识别准确度，且该方法还能够降低模型优化训练的难度，有利于降低训练成本。

技术领域

本公开涉及图像检测技术领域，尤其涉及一种金属裂纹检测方法、装置、电子设备及存储介质。

背景技术

由于生产工艺或者使用过程中长期承受交变载荷作用力，金属构件的某些薄弱部位较容易产生疲劳裂纹(局部破裂)。并且，在特定载荷或环境条件下，这些疲劳裂纹会逐渐长大，一旦扩展到临界尺寸，构件就会发生完全破坏(断裂)。

在基础设施领域中，大多数的基础设施(诸如铁轨、桥梁等)的主要组成材料为金属。为了保证基础设施的安全稳定运行，需要定期或不定期对这些金属构件进行结构健康检测。其中，这结构健康检测中的主要检测项目之一是金属裂纹的检测。

传统的金属裂纹检测，基本上是将裂纹作为无角度的矩形框检测(如图1所示)，但由于金属表面裂纹大多数是细长且有一定角度倾向的裂纹，而无角度的矩形框的框内大部分是背景区域，不能很好地表征裂纹的特征。因此，传统的金属裂纹检测方法仍存在不能很好地提取裂纹的特征信息，识别准确度较低，且优化训练难度大的问题。

发明内容

有鉴于此，本公开实施例提供了一种金属裂纹检测方法、装置、电子设备及存储介质，以解决现有的金属裂纹检测方法不能很好地提取裂纹的特征信息，识别准确度较低，且优化训练难度大的问题。

本公开实施例的第一方面，提供了一种金属裂纹检测方法，包括：

获取待检图片，将待检图片输入预设的特征提取网络，经特征提取网络的卷积层处理后，输出特征图；

使用特征提取网络的检测层对特征图进行再次特征提取和分析，得到金属裂纹检测结果，金属裂纹检测结果包括第一特征图、第二特征图、第三特征图和第四特征图；

其中，第一特征图用于表征是否存在目标物体，第二特征图用于表征目标物体的中心点相较于第一特征图的特征点的偏移量，第三特征图用于表征目标物体对应的目标框的宽高，第四特征图用于表征目标框的倾斜角度。

本公开实施例的第二方面，提供了一种金属裂纹检测装置，包括：

获取模块，被配置为获取待检图片，将待检图片输入预设的特征提取网络，经特征提取网络的卷积层处理后，输出特征图；

分析模块，被配置为使用特征提取网络的检测层对特征图进行再次特征提取和分析，得到金属裂纹检测结果，金属裂纹检测结果包括第一特征图、第二特征图、第三特征图和第四特征图；

其中，第一特征图用于表征是否存在目标物体，第二特征图用于表征目标物体的中心点相较于第一特征图的特征点的偏移量，第三特征图用于表征目标物体对应的目标框的宽高，第四特征图用于表征目标框的倾斜角度。

本公开实施例的第三方面，提供了一种电子设备，包括存储器、处理器以及存储在存储器中并且可在处理器上运行的计算机程序，该处理器执行计算机程序时实现上述方法的步骤。

本公开实施例的第四方面，提供了一种计算机可读存储介质，该计算机可读存储介质存储有计算机程序，该计算机程序被处理器执行时实现上述方法的步骤。