[发明专利]路面病害检测网络训练方法、装置、设备及存储介质

说明书

本发明公开了一种路面病害检测网络训练方法、装置、设备及存储介质，属于路面病害识别技术领域。该方法包括：获取多张路面样本图像以及对应的边缘真值矩阵和区域真值矩阵；将路面样本图像输入原始检测网络，得到对应的边缘预测矩阵和区域预测矩阵；根据边缘真值矩阵和边缘预测矩阵，得到边缘识别损失函数值，并根据区域真值矩阵和区域预测矩阵，得到区域识别损失函数值；根据两个损失函数值，调整网络参数，得到目标联合检测网络。本发明将边缘识别方式和区域识别方式相结合，从边缘网格和病害区域两个不同的角度对原始检测网络进行训练优化，使得训练后的网络可以从不同的角度进行病害检测，以结合不同角度的检测结果来保证最终的检测精度。

技术领域

本发明涉及路面病害识别技术领域，尤其涉及一种路面病害检测网络训练方法、装置、设备及存储介质。

背景技术

相关技术中，针对路面病害的采集识别可采用多种检测方法，首先采集方法主要包括线扫成像、面扫成像、激光数字成像、三维数据成像等，对应的检测装置是相机配以辅助照明的方式，而成像后的识别方法为基于灰度识别方法和深度学习方法，基于灰度图像中不同成分的频率进行识别检测。

但是，在这种方式中，路面的水渍污迹、标志标线、光照不均及路面阴影等情况都会对病害识别造成极大干扰，且该方法参考的标准比较单一，相应的可以检测的病害种类也比较单一，故现阶段无法从不同的角度对路面病害进行识别检测，检测精度较低。

发明内容

本发明的主要目的在于提供一种路面病害检测网络训练方法、装置、设备及存储介质，旨在解决现有技术中路面病害检测精度较低的技术问题。

根据本发明的第一方面，提供了一种路面病害检测网络训练方法，所述方法包括：

获取多张路面样本图像以及对应的多组边缘真值矩阵和区域真值矩阵；所述边缘真值矩阵的每一元素对应于一个边缘网格，所述边缘真值矩阵中每一元素的取值表示对应的边缘网格中是否具有指定路面病害，所述区域真值矩阵中每一行元素对应于一个病害区域，所述区域真值矩阵每一行中的元素取值表示对应的病害区域的外接矩形坐标；其中，所述边缘网格由所述路面样本图像按照固定尺寸分割得到，所述病害区域依据所述路面样本图像中属于指定路面病害的区域的最小外接矩形得到；

将多张所述路面样本图像输入原始检测网络，得到对应的边缘预测矩阵和区域预测矩阵；

根据所述边缘真值矩阵和所述边缘预测矩阵，得到边缘识别损失函数值，并根据所述区域真值矩阵和所述区域预测矩阵，得到区域识别损失函数值；

根据所述边缘识别损失函数值和所述区域识别损失函数值，调整所述原始检测网络的网络参数，得到目标联合检测网络。

可选地，所述根据所述边缘真值矩阵和所述边缘预测矩阵，得到边缘识别损失函数值，并根据所述区域真值矩阵和所述区域预测矩阵，得到区域识别损失函数值，包括：

根据第一公式、所述边缘真值矩阵和所述边缘预测矩阵，得到所述边缘识别损失函数值，并根据第二公式、所述区域真值矩阵和所述区域预测矩阵，得到所述区域识别损失函数值；其中，所述第一公式为：

其中，loss1为所述边缘识别损失函数值，为所述边缘真值矩阵的矩阵向量，为所述边缘预测矩阵的矩阵向量，ε为预设常数以避免分母为0；

所述第二公式为：

其中，loss2为所述区域识别损失函数值，为所述区域真值矩阵的矩阵向量，为所述区域预测矩阵的矩阵向量。

可选地，所述根据所述边缘识别损失函数值和所述区域识别损失函数值，调整所述原始检测网络的网络参数，得到目标联合检测网络，包括：

根据所述边缘识别损失函数值和所述区域识别损失函数值，得到总损失函数值；