本申请提供了一种地下病害检测方法、装置、电子设备和存储介质，其中，该方法包括：获取回波图像后，根据回波图像的标注数据生成标注图像数据，并对其进行数据增强处理，得到多个数据样本，将其输入初始病害检测模型中并得到初始预测结果，根据初始预测结果和标注数据对初始预测结果进行正负样本分配，得到正负样本。通过正负样本及标注数据确定模型损失，根据损失对模型进行迭代修正，得到目标病害检测模型，基于目标病害检测模型进行地下病害检测。本申请通过基于初始预测结果，生成正负样本，使损失计算量减小，进而根据损失迭代修正得到目标病害检测模型，避免人工标注所耗费的大量人力物力资源，且不需要检测人员具备地质知识。知识。知识。
【技术实现步骤摘要】
地下病害检测方法、装置、电子设备和存储介质
[0001]本申请涉及地下病害检测
，具体而言，涉及一种地下病害检测方法、装置、电子设备和存储介质。
技术介绍
[0002]随着地下交通的发展，病害可能会严重威胁地下交通安全。因此，检测地下病害变的尤为重要。
[0003]现有技术中，可以通过探地雷达对地下病害进行监测，并生成回波图像。再将回波图像输入计算机中，并通过传统人工对回波图像中的病害进行识别和标注，从而检测出地下病害。
[0004]但是，人工标注病害信息需要耗费大量的人力物力资源，且对回波图像进行识别需要检测人员具备一定的地质知识，并对探地雷达有一定了解，而此类检测人员较少，因此对地下的检测造成阻碍。
技术实现思路
[0005]本申请的目的在于，针对上述现有技术中的不足，提供一种地下病害检测方法、装置、电子设备和存储介质，以解决现有技术中人工标注需要耗费大量人力物力资源，且需要检测人员具备地质知识的问题。
[0006]为实现上述目的，本申请实施例采用的技术方案如下：
[0007]第一方面，本申请一实施例提供了一种地下病害检测方法，所述方法包括：
[0008]获取雷达对进行扫描所得到的回波图像；
[0009]根据用户对所述回波图像中的病害目标进行标注的标注数据，生成标注图像数据，标注图像数据包括：病害图像以及标注数据；
[0010]根据所述标注图像数据进行数据增强处理，得到多个数据样本；
[0011]将各所述数据样本输入初始病害检测模型，得到所述初始病害检测模型输出的所述数据样本对应的初始预测结果；
[0012]根据所述数据样本对应的初始预测结果以及所述数据样本对应的标注数据，对所述初始预测结果进行正负样本分配，得到所述数据样本对应的至少一个正样本以及多个负样本；
[0013]根据所述至少一个正样本、所述多个负样本以及所述数据样本对应的标注数据，确定所述初始病害检测模型的损失，并根据所述初始病害检测模型的损失对所述初始病害检测模型进行迭代修正，得到目标病害检测模型；
[0014]基于所述目标病害检测模型进行地下病害检测。
[0015]可选的，所述根据所述数据样本对应的初始预测结果以及所述数据样本对应的标注数据，对所述初始预测结果进行正负样本分配，得到所述数据样本对应的至少一个正样本以及多个负样本，包括：
[0016]根据所述数据样本对应的初始预测结果以及所述数据样本对应的标注数据，确定所述数据样本对应的初始预测结果中各预测框的准确度信息，其中，所述初始预测结果中包括多个预测框，所述标注数据中包括至少一个真实框；
[0017]根据各预测框的准确度信息确定正样本数；
[0018]根据所述正样本数和所述初始预测结果确定所述数据样本对应的至少一个正样本，并将所述初始预测结果的预测框中除所述正样本之外的预测框作为所述多个负样本。
[0019]可选的，所述根据所述数据样本对应的初始预测结果以及所述数据样本对应的标注数据，确定所述数据样本对应的初始预测结果中各预测框的准确度信息，包括：
[0020]确定所述初始预测结果的各预测框与所述真实框之间的重合度信息以及种类预测准确度信息；
[0021]确定所述真实框的中心点与所述各预测框的中心点的距离；
[0022]根据所述重合度信息、所述种类预测准确度信息以及所述距离，确定各预测框的准确度信息。
[0023]可选的，所述根据各预测框的准确度信息确定正样本数，包括：
[0024]根据各预测框的准确度信息，确定预设个数的待选预测框；
[0025]根据各待选预测框对应的重合度信息，确定各待选预测框与所述真实框的面积交并比；
[0026]将各待选预测框与所述真实框的面积交并比相加并取整，得到所述正样本数。
[0027]可选的，所述根据所述至少一个正样本、所述多个负样本以及所述数据样本对应的标注数据，确定所述初始病害检测模型的损失，包括：
[0028]根据所述正样本中的预测框和所述标注数据，确定交叠损失；
[0029]根据所述正样本中的置信度、负样本中的置信度和所述标注数据，确定置信度损失；
[0030]根据所述正样本的病害种类和所述标注数据，确定病害种类损失；
[0031]根据所述交叠损失、所述置信度损失和所述病害种类损失，确定所述初始病害检测模型的损失。
[0032]可选的，所述根据所述标注图像数据进行数据增强处理，得到多个数据样本，包括：
[0033]从所述标注图像数据中选择第一预设数量的第一待增强数据样本，并基于所述第一待增强数据样本按照第一数据增强方式进行数据增强，得到第一增强数据样本，其中，所述第一数据增强方式包括：病害图像裁剪、旋转、随机拼接；
[0034]从所述第一增强数据样本中选择第二预设数量的第二待增强数据样本，并基于所述第二待增强数据样本按照第二数据增强方式进行数据增强，得到第二增强数据样本；
[0035]将所述标注图像数据、所述第一增强数据样本和所述第二增强数据样本组合为所述多个数据样本。
[0036]可选的，所述基于所述第二待增强数据样本按照第二数据增强方式进行数据增强，得到第二增强数据样本，包括：
[0037]将多个所述第二待增强数据样本的病害图像按照预设的透明度进行融合，得到一个所述第二增强数据样本的病害图像，并将多个所述第二待增强数据样本的标注数据作为
所述第二增强数据样本的标注数据。
[0038]第二方面，本申请另一实施例提供了一种地下病害检测装置，所述装置包括：
[0039]获取模块，用于获取雷达对进行扫描所得到的回波图像；
[0040]生成模块，用于根据用户对所述回波图像中的病害目标进行标注的标注数据，生成标注图像数据，标注图像数据包括：病害图像以及标注数据；
[0041]数据增强模块，用于根据所述标注图像数据进行数据增强处理，得到多个数据样本；
[0042]输入模块，用于将各所述数据样本输入初始病害检测模型，得到所述初始病害检测模型输出的所述数据样本对应的初始预测结果；
[0043]分配模块，用于根据所述数据样本对应的初始预测结果以及所述数据样本对应的标注数据，对所述初始预测结果进行正负样本分配，得到所述数据样本对应的至少一个正样本以及多个负样本；
[0044]修正模块，用于根据所述至少一个正样本、所述多个负样本以及所述数据样本对应的标注数据，确定所述初始病害检测模型的损失，并根据所述初始病害检测模型的损失对所述初始病害检测模型进行迭代修正，得到目标病害检测模型；
[0045]检测模块，用于基于所述目标病害检测模型进行地下病害检测。
[0046]可选的，所述分配模块具体包括：
[0047]根据所述数据样本对应的初始预测结果以及所述数据样本对应的标注数据，确定所述数据样本对应的初始预测结果中各预测框的准确度信息，其中本文档来自技高网...
【技术保护点】
【技术特征摘要】
1.一种地下病害检测方法，其特征在于，包括：获取雷达对进行扫描所得到的回波图像；根据用户对所述回波图像中的病害目标进行标注的标注数据，生成标注图像数据，标注图像数据包括：病害图像以及标注数据；根据所述标注图像数据进行数据增强处理，得到多个数据样本；将各所述数据样本输入初始病害检测模型，得到所述初始病害检测模型输出的所述数据样本对应的初始预测结果；根据所述数据样本对应的初始预测结果以及所述数据样本对应的标注数据，对所述初始预测结果进行正负样本分配，得到所述数据样本对应的至少一个正样本以及多个负样本；根据所述至少一个正样本、所述多个负样本以及所述数据样本对应的标注数据，确定所述初始病害检测模型的损失，并根据所述初始病害检测模型的损失对所述初始病害检测模型进行迭代修正，得到目标病害检测模型；基于所述目标病害检测模型进行地下病害检测。2.根据权利要求1所述的地下病害检测方法，其特征在于，所述根据所述数据样本对应的初始预测结果以及所述数据样本对应的标注数据，对所述初始预测结果进行正负样本分配，得到所述数据样本对应的至少一个正样本以及多个负样本，包括：根据所述数据样本对应的初始预测结果以及所述数据样本对应的标注数据，确定所述数据样本对应的初始预测结果中各预测框的准确度信息，其中，所述初始预测结果中包括多个预测框，所述标注数据中包括至少一个真实框；根据各预测框的准确度信息确定正样本数；根据所述正样本数和所述初始预测结果确定所述数据样本对应的至少一个正样本，并将所述初始预测结果的预测框中除所述正样本之外的预测框作为所述多个负样本。3.根据权利要求2所述的地下病害检测方法，其特征在于，所述根据所述数据样本对应的初始预测结果以及所述数据样本对应的标注数据，确定所述数据样本对应的初始预测结果中各预测框的准确度信息，包括：确定所述初始预测结果的各预测框与所述真实框之间的重合度信息以及种类预测准确度信息；确定所述真实框的中心点与所述各预测框的中心点的距离；根据所述重合度信息、所述种类预测准确度信息以及所述距离，确定各预测框的准确度信息。4.根据权利要求2所述的地下病害检测方法，其特征在于，所述根据各预测框的准确度信息确定正样本数，包括：根据各预测框的准确度信息，确定预设个数的待选预测框；根据各待选预测框对应的重合度信息，确定各待选预测框与所述真实框的面积交并比；将各待选预测框与所述真实框的面积交并比相加并取整，得到所述正样本数。5.根据权利要求1所述的地下病害检测方法，其特征在于，所述根据所述至少一个正样本、所述多个负样本以及所述数据样本对应的标注数据，确定所述初始病害检测模型的损
失，包括：根据所述正样本中的预测框和所述标注数据，确定交叠损失；根据所述正样本中的置信度、...
【专利技术属性】
技术研发人员：许明，张弓，郑睿博，陈星，薛海红，王斌，李开鹏，陈培尧，
申请(专利权)人：中煤西安地下空间科技发展有限公司，
类型：发明
国别省市：
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。