一种基于伪缺陷空间生成的电力设备缺陷检测方法
摘要:
本发明提出了一种基于伪缺陷空间生成的电力设备缺陷检测方法,以正常图片为素材人为伪造出局部的不规则性,生成一种伪缺陷图像类别,在对正常图像、缺陷图像、伪缺陷图像进行特征提取和分类网络的基础上,利用对比学习对特征空间区域解耦,对伪缺陷图像特征、正常图像特征、缺陷图像特征嵌入约束,增强难分样本中复杂背景的抗干扰能力,提高模型对于电力设备隐蔽缺陷的识别精度;利用电力设备缺陷模型预判得到分类损失并依据分类损失获取边缘伪缺陷样本,在进行若干次迭代训练后利用边缘伪缺陷样本引入弱监督缺陷遮盖策略进行训练,缓解过拟合,帮助模型突破训练瓶颈,获得最佳模型参数并固定,使用该电力设备缺陷检测模型对待检测图像进行检测。
0/0