[发明专利]隧道抛洒物检测的方法、装置、电子设备及存储介质

权利要求书

1.一种隧道抛洒物检测的方法，其特征在于，包括：

获取隧道的视频数据；

对所述视频数据采用平均帧处理，得到隧道平均帧图像；

将所述隧道平均帧图像输入至训练好的分割模型，得到所述隧道平均帧图像中不同类型抛洒物对应区域的分割结果；

根据所述分割结果确定所述抛洒物的位置信息；

其中，所述训练好的分割模型为利用不同隧道平均帧图像以及不同隧道平均帧图像对应的语义分割真值图进行训练后得到。

2.根据权利要求1所述的隧道抛洒物检测的方法，其特征在于，在将所述隧道平均帧图像输入至训练好的分割模型，得到所述隧道平均帧图像中不同类型抛洒物对应区域的分割结果之前，还包括：

获取隧道平均帧训练图像以及语义分割真值图；

提取所述隧道平均帧训练图像的梯度信息，得到第一特征图；

对所述第一特征图进行预处理，得到第二特征图；

基于E-Net网络对所述第二特征图编码后解码，得到语义分割预测图；

根据所述语义分割真值图以及所述语义分割预测图确定损失值；

根据所述损失值更新所述分割模型的参数，得到训练好的分割模型。

3.根据权利要求2所述的隧道抛洒物检测的方法，其特征在于，所述对所述第一特征图进行预处理，得到第二特征图，包括：

对所述第一特征图卷积得到第一下采样特征图；

对所述第一特征图最大池化得到第二下采样特征图；

将所述第一下采样特征图以及所述第二下采样特征图进行拼接，得到第二特征图。

4.根据权利要求2所述的隧道抛洒物检测的方法，其特征在于，所述网络包含第一编码器、第二编码器、第一解码器、第二解码器以及分类器，所述基于E-Net网络对所述第二特征图编码后解码，得到语义分割预测图，包括：

采用所述第一编码器对所述第二特征图编码后采用坐标注意力机制，得到第三特征图；

采用所述第二编码器对所述第三特征图编码后采用坐标注意力机制，得到第四特征图；

采用所述第一解码器对所述第四特征图解码后采用坐标注意力机制，得到第五特征图；

采用所述第二解码器对所述第五特征图解码后采用所述分类器分类，得到语义分割预测图。

5.根据权利要求1所述的隧道抛洒物检测的方法，其特征在于，所述根据所述分割结果确定所述抛洒物的位置信息，包括：

获取二值图模板；

根据所述分割结果确定待检测二值图；

将所述待检测二值图中每个像素点的像素值分别与所述二值图模板中对应的像素点的像素值相减，得到二值结果图；

根据所述二值结果图确定所述抛洒物的位置信息。

6.一种隧道抛洒物检测的装置，其特征在于，包括：

获取模块，用于获取隧道的视频数据；

处理模块，用于对所述视频数据采用平均帧处理，得到隧道平均帧图像；将所述隧道平均帧图像输入至训练好的分割模型，得到所述隧道平均帧图像中不同类型抛洒物对应区域的分割结果；根据所述分割结果确定所述抛洒物的位置信息；其中，所述训练好的分割模型为利用不同隧道平均帧图像以及不同隧道平均帧图像对应的语义分割真值图进行训练后得到。

7.根据权利要求6所述的隧道抛洒物检测的装置，其特征在于，所述处理模块具体用于：

获取二值图模板；

根据所述分割结果确定待检测二值图；

将所述待检测二值图中每个像素点的像素值分别与所述二值图模板中对应的像素点的像素值相减，得到二值结果图；

根据所述二值结果图确定所述抛洒物的位置信息。

8.一种计算机程序产品，包括计算机程序，其特征在于，所述计算机程序被处理器执行时实现如权利要求1至5任一项所述方法的步骤。

9.一种电子设备，包括存储器、处理器及存储在所述存储器上并可在所述处理器上运行的计算机程序，其特征在于，所述处理器执行所述程序时实现如权利要求1至5任一项所述方法的步骤。

10.一种非暂态计算机可读存储介质，其上存储有计算机程序，其特征在于，所述计算机程序被处理器执行时实现如权利要求1至5任一项所述方法的步骤。