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

权利要求书

1.一种图像检测方法，其特征在于，所述方法包括：

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

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

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

2.根据权利要求1所述的方法，其特征在于，所述对所述至少一张第一掩码图进行检测，确定所述高铁接触网是否存在目标部件缺失，包括：

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

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

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

3.根据权利要求2所述的方法，其特征在于，所述每张第一掩码图中包括进行像素填充得到的像素填充区域，所述对所述至少一张第一掩码图进行目标检测，得到所述至少一张第一掩码图中每张第一掩码图对应的参考预测框，包括：

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

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

4.根据权利要求2所述的方法，其特征在于，所述根据所述第一目标预测框，确定所述高铁接触网是否存在目标部件缺失，包括：

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

根据所述位置信息从所述待检测图像中截取出待分类区域图像；

对所述待分类区域图像进行检测，确定所述高铁接触网是否存在目标部件缺失。

5.根据权利要求1-4任一项所述的方法，其特征在于，所述获取高铁接触网的待检测图像，包括：

对所述高铁接触网的原始图像进行目标检测，得到第一目标检测结果；

若所述第一目标检测结果中包括至少一个第二预测框，则从所述至少一个第二预测框中确定出第二目标预测框，根据所述第二目标预测框和预设的网格，得到所述待检测图像；

若所述第一目标检测结果中不包括第二预测框，则根据预设的矩形框和所述网格，得到所述待检测图像。

6.根据权利要求5所述的方法，其特征在于，所述根据所述第二目标预测框和预设的网格，得到所述待检测图像，包括：

根据所述网格的尺寸和所述第二目标预测框的宽度或高度，计算得到第一缩放系数；

按照所述第一缩放系数对所述原始图像进行缩放，得到所述待检测图像。

7.根据权利要求5所述的方法，其特征在于，所述根据预设的矩形框和所述网格，得到所述待检测图像，包括：

根据所述网格的尺寸和所述矩形框的宽度或高度，计算得到第二缩放系数；

按照所述第二缩放系数对所述原始图像进行缩放，得到所述待检测图像。

8.根据权利要求3-7任一项所述的方法，其特征在于，所述对所述每张第一掩码图进行目标检测，得到所述每张第一掩码图对应的第一预测框，包括：

执行至少一次卷积、门控卷积和权重系数相乘的操作，将得到的融合特征图作为目标特征图；

基于所述目标特征图对所述每张第一掩码图进行分类预测，得到所述第一预测框；

其中，至少一次卷积、门控卷积和权重系数相乘的操作，包括：

对所述每张第一掩码图或者上次操作得到的特征图进行卷积，得到第一特征图；

对所述每张第一掩码图或者上次操作得到的特征图进行门控卷积，得到第二特征图，所述第二特征图与所述第一特征图的尺寸相同，所述第二特征图中每个位置的值用于表示所述第一特征图中对应位置的权重系数；

将所述第一特征图中每个位置的值与所述第二特征图中对应的权重系数相乘，得到融合特征图。