[发明专利]一种防震锤缺陷检测方法、装置、设备和存储介质

权利要求书

1.一种防震锤缺陷检测方法，其特征在于，包括：

获取待检测图像在特征提取后得到的至少一个尺度下的缩放图像；其中，所述待检测图像中包括待检测防震锤；

对各所述缩放图像进行处理，得到至少一个尺度下的缩放特征图；

对至少一个所述缩放图像进行特征增强，得到残差增强图；

对所述残差增强图和各尺度下的所述缩放特征图进行融合，得到融合特征图；

根据所述融合特征图，确定所述待检测图像中的待检测防震锤的缺陷情况。

2.根据权利要求1所述的方法，其特征在于，所述对至少一个所述缩放图像进行特征增强，得到残差增强图，包括：

将至少一个尺度下的缩放图像作为当前特征图；

针对各所述当前特征图，对该当前特征图进行池化处理，得到中间特征图；

根据所述中间特征图，确定该当前特征图的残差增强图。

3.根据权利要求2所述的方法，其特征在于，所述对该当前特征图进行池化处理，得到中间特征图，包括：

对该当前特征图在至少一个尺度下进行池化处理，得到各尺度下的中间特征图；

相应的，所述根据所述中间特征图，确定该当前特征图的残差增强图，包括：

对各尺度下的中间特征图进行降维处理，得到降维特征图；

对各所述降维特征图上采样后进行特征融合，得到该当前特征图的残差增强图。

4.根据权利要求3所述的方法，其特征在于，所述对各所述降维特征图上采样后进行特征融合，得到该当前特征图的残差增强图，包括：

对各所述降维特征图上采样，得到上采样特征图；

根据各池化处理对应池化权重，对各所述上采样特征图进行加权融合，得到该当前特征图的残差增强图。

5.根据权利要求1所述的方法，其特征在于，所述对所述残差增强图和各尺度下的所述缩放特征图进行融合，得到融合特征图，包括：

将所述残差增强图与相应缩放特征图进行融合，以更新所述缩放特征图；

对各缩放特征图进行尺度调整，得到相应深层特征图；

对各所述深层特征图进行融合，得到所述融合特征图。

6.根据权利要求1-5任一项所述的方法，其特征在于，所述根据所述融合特征图，确定所述待检测图像中的待检测防震锤的缺陷情况，包括：

基于区域候选网络RPN，根据所述融合特征图，确定所述待检测图像中的待检测防震锤的缺陷情况。

7.一种防震锤缺陷检测装置，其特征在于，包括：

图像获取模块，用于获取待检测图像在特征提取后得到的至少一个尺度下的缩放图像；其中，所述待检测图像中包括待检测防震锤；

缩放特征图获取模块，用于对各所述缩放图像进行处理，得到至少一个尺度下的缩放特征图；

残差增强模块，用于对至少一个所述缩放图像进行特征增强，得到残差增强图；

特征融合模块，用于对所述残差增强图和各尺度下的所述缩放特征图进行融合，得到融合特征图；

缺陷检测模块，用于根据所述融合特征图，确定所述待检测图像中的待检测防震锤的缺陷情况。

8.根据权利要求7所述的装置，其特征在于，所述残差增强模块包括：

当前特征图确定单元，用于将至少一个尺度下的缩放图像作为当前特征图；

中间特征图确定单元，用于针对各当前特征图，对该当前特征图进行池化处理，得到中间特征图；

残差增强图确定单元，用于根据中间特征图，确定该当前特征图的残差增强图。

9.一种电子设备，其特征在于，包括：

一个或多个处理器；

存储器，用于存储一个或多个程序；

当所述一个或多个程序被所述一个或多个处理器执行，使得所述一个或多个处理器实现如权利要求1-6任一项所述的一种防震锤缺陷检测方法。

10.一种计算机可读存储介质，其上存储有计算机程序，其特征在于，该程序被处理器执行时实现如权利要求1-6任一项所述的一种防震锤缺陷检测方法。