本发明涉及图像处理,特别是涉及一种配电线路的缺陷检测方法、装置、电子设备和介质。
背景技术:
1、相关技术中,目标检测(detr,detection transformer)模型能够用于将图像中待检测图像与背景区分开,并预测图像中待检测对象的位置和类别。detr模型的实现原理简单,且无需复杂的后处理逻辑,能够直接输出检测结果,但是detr模型存在检测效果差,且模型训练速度慢的问题。
技术实现思路
1、基于此,有必要针对上述技术问题,提供一种能够解决应用detr模型的图像检测效果差、训练速度慢的配电线路的缺陷检测方法、装置、电子设备和介质。
2、一种配电线路的缺陷检测方法,包括以下步骤:
3、获取经过预处理的配电线路图片;
4、对经过预处理的配电线路图片进行特征提取,获得第一特征图;
5、将所述第一特征图输入目标检测模型的编码层,获得第二特征图和第一位置编码;
6、将所述第二特征图和第一位置编码输入到所述目标检测模型的解码层,获得多个检测结果,其中,所述目标检测模型的解码层采用模型学习随机初始化的内容查询向量以及通过去噪查询向量辅助训练的方式来更新匹配查询向量的方式进行训练,并在完成模型训练后去除所述去噪查询向量;
7、根据所述多个检测结果确定所述配电线路图片的缺陷检测结果。
8、一种配电线路的缺陷检测装置,包括:
9、第一获取模块,用于获取经过预处理的配电线路图片;
10、第二获取模块,用于对经过预处理的配电线路图片进行特征提取,获得第一特征图;
11、第三获取模块,用于将所述第一特征图输入目标检测模型的编码层,获得第二特征图和第一位置编码;
12、检测模块,用于将所述第二特征图和第一位置编码输入到所述目标检测模型的解码层,获得多个检测结果,其中,所述目标检测模型的解码层采用模型学习随机初始化的内容查询向量以及通过去噪查询向量辅助训练的方式来更新匹配查询向量的方式进行训练,并在完成模型训练后去除所述去噪查询向量;
13、确定模块,用于根据所述多个检测结果确定所述配电线路图片的缺陷检测结果。
14、一种电子设备,包括存储器和处理器,存储器存储有计算机程序,处理器执行计算机程序时实现上述配电线路的缺陷检测方法的步骤。
15、一种计算机可读存储介质,其上存储有计算机程序,计算机程序被处理器执行时实现上述配电线路的缺陷检测方法的步骤。
16、上述配电线路的缺陷检测方法、装置、电子设备和介质,采用模型学习随机初始化的内容查询向量以及通过去噪查询向量辅助训练的方式来更新匹配查询向量的方式进行训练,并在完成模型训练后去除去噪查询向量,从而使目标检测模型在训练的过程中能够更快速的收敛,同时提升目标检测模型的检测性能,提高检测的精确度。
1.一种配电线路的缺陷检测方法,其特征在于,包括:
2.根据权利要求1所述的方法,其特征在于,在所述目标检测模型的解码层采用模型学习随机初始化的内容查询向量以及通过去噪查询向量辅助训练的方式来更新匹配查询向量的方式进行训练时,所述方法包括:
3.根据权利要求2所述的方法,其特征在于,在所述将所述第四特征图和所述第二位置编码输入至所述目标检测模型中的解码层以对所述目标检测模型中的解码层进行训练时,所述方法包括:
4.根据权利要求2所述的方法,其特征在于,所述将所述第四特征图和所述第二位置编码输入至所述目标检测模型中的解码层以对所述目标检测模型中的解码层进行训练,方法包括:
5.根据权利要求2所述的方法,其特征在于,在所述将所述第四特征图和所述第二位置编码输入至所述目标检测模型中的解码层以对所述目标检测模型中的解码层进行训练之前,包括:
6.根据权利要求2所述的方法,其特征在于,所述将所述第四特征图和所述第二位置编码输入至所述目标检测模型中的解码层以对所述目标检测模型中的解码层进行训练,包括:
7.根据权利要求6所述的方法,其特征在于,在进入下一个计算循环之前,所述方法还包括:
8.一种配电线路的缺陷检测装置,其特征在于,包括:
9.一种电子设备,包括存储器和处理器,所述存储器存储有计算机程序,其特征在于,所述处理器执行所述计算机程序时实现权利要求1至7中任一项所述方法的步骤。
10.一种计算机可读存储介质,其上存储有计算机程序,其特征在于,所述计算机程序被处理器执行时实现权利要求1至7中任一项所述方法的步骤。