[发明专利]一种螺栓松动检测方法、装置及存储介质

说明书

本发明公开了一种螺栓松动检测方法、装置及存储介质，其中方法包括获取螺栓图像；对螺栓图像进行图像增强处理；通过检测模型对增强图像提取特征得到多尺度的预测特征，将尺度最小的预测特征池化，将多种池化特征和尺度最小的预测特征拼接，根据拼接特征得到螺栓松动角度；根据螺栓松动角度得到判断结果；通过图像增强增加数据多样性，增加图像训练数，能使用较少的数据集对检测模型训练出更高的检测精度；通过检测模型对预测特征池化和拼接，增加了特征深度，有利于对检测模型调参优化以得到更准确的检测模型。

技术领域

本发明涉及图像检测领域，特别是一种螺栓松动检测方法、装置及存储介质。

背景技术

现有的螺栓松动检测技术通常采用在螺栓固定区域装配传感器进行检测。但精密的传感器不仅成本极高，并且传感器容易受到干扰导致检测结果差异大。而人工检测则人力成本高，检测效率低下。

发明内容

本发明的目的在于至少解决现有技术中存在的技术问题之一，提供一种螺栓松动检测方法、装置及存储介质。

本发明解决其问题所采用的技术方案是：

本发明的第一方面，一种螺栓松动检测方法，包括：

获取螺栓图像；

对所述螺栓图像进行图像增强处理，得到增强图像；

通过检测模型对所述增强图像进行特征提取，得到多种尺度不同的预测特征，将尺度最小的所述预测特征进行池化操作得到多种池化特征，将多种所述池化特征和尺度最小的所述预测特征拼接得到拼接特征，根据所述拼接特征得到螺栓松动角度；

比较所述螺栓松动角度与角度阈值的数值大小，得到判断结果。

根据本发明的第一方面，所述对所述螺栓图像进行图像增强处理，得到增强图像的步骤，具体为：将多张不同的所述螺栓图像拼接得到所述增强图像，其中每次拼接的所述螺栓图像的数量相同。

根据本发明的第一方面，所述将尺度最小的所述预测特征进行池化操作得到多种池化特征，将多种所述池化特征和尺度最小的所述预测特征拼接得到拼接特征的步骤，具体为：将尺度最小的所述预测特征输入三个最大池化层进行最大池化下采样得到三种池化特征，将三种所述池化特征和尺度最小的所述预测特征拼接得到拼接特征，其中三个所述最大池化层的池化核大小分别为3*3、5*5和9*9，三个所述最大池化层的步距为1。

根据本发明的第一方面，在所述将尺度最小的所述预测特征进行池化操作得到多种池化特征，将多种所述池化特征和尺度最小的所述预测特征拼接得到拼接特征的步骤中，还得到多个权重值；

所述螺栓松动检测方法还包括以下步骤：通过多个所述权重值对所述检测模型的超参数进行调整。

本发明的第二方面，一种螺栓松动检测装置，包括:

输入模块，用于获取螺栓图像；

图像增强模块，用于对所述螺栓图像进行图像增强处理，得到增强图像；

检测模块，用于对所述增强图像进行特征提取，得到多种尺度不同的预测特征，将尺度最小的所述预测特征进行池化操作得到多种池化特征，将多种所述池化特征和尺度最小的所述预测特征拼接得到拼接特征，根据所述拼接特征得到螺栓松动角度；

输出模块，用于比较所述螺栓松动角度与角度阈值的数值大小，得到判断结果。

根据本发明的第二方面，在所述图像增强模块中，将多张不同的所述螺栓图像拼接得到所述增强图像，其中每次拼接的所述螺栓图像的数量相同。