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

说明书

本申请实施例提供一种图像检测方法、装置、电子设备及存储介质，其中，该图像检测方法包括：获取高铁接触网的待检测图像；基于预设的填充方式对待检测图像进行像素填充，得到至少一张第一掩码图；对至少一张第一掩码图进行检测，确定高铁接触网是否存在目标部件缺失。本申请实施例有利于降低接触网螺母、螺栓和开口销缺失检测的难度。

技术领域

本申请涉及图像检测技术领域，具体涉及一种图像检测方法、装置、电子设备及存储介质。

背景技术

接触网是铁路电气化工程中的重要组成部分，对接触网进行定期巡检是乘客安全出行的有力保障。在工业领域中，尤其是是高铁领域，接触网螺母、螺栓、开口销缺失检测技术已经有了较为广泛的应用，其总体思路是通过缺失检测模型，进行螺母、螺栓、开口销的粗定位，利用螺母、螺栓、开口销的纹理特征进行缺失检测。但是，相比正常的螺母、螺栓、开口销数据，螺母、螺栓、开口销缺失的数据数量十分少，这使得缺失检测的建模难度相对较大；另一方面，螺母、螺栓、开口销存在的场景不同，其呈现的特征也不相同，加大了缺失检测中的特征提取难度。可见，目前的接触网螺母、螺栓、开口销缺失检测较为复杂，存在检测难度大的问题。

发明内容

本申请实施例提供一种图像检测方法、装置、电子设备及存储介质，有利于降低接触网螺母、螺栓和开口销缺失检测的难度。

本申请实施例的第一方面提供一种图像检测方法，该方法包括：

获取高铁接触网的待检测图像；

基于预设的填充方式对待检测图像进行像素填充，得到至少一张第一掩码图；

对至少一张第一掩码图进行检测，确定高铁接触网是否存在目标部件缺失。

可以看出，本申请实施例通过获取高铁接触网的待检测图像；基于预设的填充方式对待检测图像进行像素填充，得到至少一张第一掩码图；对至少一张第一掩码图进行检测，以确定高铁接触网是否存在目标部件缺失。这样通过像素填充的方式，有利于降低缺失检测中对螺母、螺栓和开口销等目标部件的纹理特征的依赖，以通过上下文信息进行缺失检测，从而有利于降低接触网螺母、螺栓和开口销缺失检测的难度。

结合第一方面，在一个可能的实现方式中，对至少一张第一掩码图进行检测，确定高铁接触网是否存在目标部件缺失，包括：

对至少一张第一掩码图进行目标检测，得到至少一张第一掩码图中每张第一掩码图对应的参考预测框；

对参考预测框进行融合，得到第一目标预测框；

根据第一目标预测框，确定高铁接触网是否存在目标部件缺失。

本示例中，由于第一目标预测框是对参考预测框进行融合得到的预测框，说明第一目标预测框所框定的区域最有可能存在螺母、螺栓、开口销等目标部件，利用该区域进行缺失检测，有利于提高缺失检测的准确度。

结合第一方面，在一个可能的实现方式中，每张第一掩码图中包括进行像素填充得到的像素填充区域，对至少一张第一掩码图进行目标检测，得到至少一张第一掩码图中每张第一掩码图对应的参考预测框，包括：

对每张第一掩码图进行目标检测，得到每张第一掩码图对应的第一预测框；

将每张第一掩码图中落在像素填充区域的第一预测框，确定为参考预测框。

本示例中，将每张第一掩码图中整个落在像素填充区域的第一预测框作为参考预测框，有利于避免后续预测时用到非像素填充区域的物体纹理信息，从而降低对物体纹理信息的依赖。

结合第一方面，在一个可能的实现方式中，根据第一目标预测框，确定高铁接触网是否存在目标部件缺失，包括：

确定第一目标预测框在待检测图像中的位置信息；