【技术实现步骤摘要】
两分支神经网络缺陷检测方法、装置、设备、介质和产品
[0001]本申请涉及缺陷检测
,特别是涉及一种基于注意力机制的两分支卷积神经网络电力设备表面缺陷检测方法、装置、计算机设备、存储介质和计算机程序产品。
技术介绍
[0002]随着电力领域的快速发展,对电力设备的表面缺陷检测技术成为了重要的检测技术。
[0003]传统技术中,通常是通过频域滤波检测方法将待检测图像首先变换到频域,提取缺陷分量后再次反变换到时域,通过Blob分析获得缺陷的具体位置。但该技术检测电力设备的表面缺陷的准确性较低。
技术实现思路
[0004]基于此,有必要针对上述技术问题,提供一种基于注意力机制的两分支卷积神经网络电力设备表面缺陷检测方法、装置、计算机设备、计算机可读存储介质和计算机程序产品。
[0005]第一方面,本申请提供了一种基于注意力机制的两分支卷积神经网络电力设备表面缺陷检测方法。所述方法包括:
[0006]基于卷积层构建得到粗特征提取网络;
[0007]基于自编码器构建得到形状特征提取 ...
【技术保护点】
【技术特征摘要】
1.一种基于注意力机制的两分支卷积神经网络电力设备表面缺陷检测方法,其特征在于,所述方法包括:基于卷积层构建得到粗特征提取网络;基于自编码器构建得到形状特征提取分支网络;基于至少两个残差注意力模块,构建得到纹理特征提取分支网络;基于所述形状特征提取分支网络和纹理特征提取分支网络,构建得到两分支特征提取网络;基于最大池化和conv层,构建得到决策网络;基于将所述粗特征提取网络、两分支特征提取网络和决策网络依序连接,构建得到电力设备表面缺陷检测模型;获取电力设备的待检测图像,利用所述电力设备表面缺陷检测模型对所述待检测图像进行检测得到所述电力设备的图像检测结果。2.根据权利要求1所述的方法,其特征在于,所述方法还包括:基于特征提取层、映射层、构造层、Sigmod激活函数和注意力掩码,得到所述残差注意力模块。3.根据权利要求1所述的方法,其特征在于,所述基于将所述粗特征提取网络、两分支特征提取网络和决策网络依序连接,构建得到电力设备表面缺陷检测模型,包括:将所述粗特征提取网络、两分支特征提取网络和决策网络依序连接,构建得到电力设备表面缺陷检测初始模型;通过缺陷检测公开数据库对所述电力设备表面缺陷检测初始模型进行训练,构建得到所述电力设备表面缺陷检测模型。4.根据权利要求3所述的方法,其特征在于,所述训练的损失函数基于平均绝对误差构建。5.根据权利要求1所述的方法,其特征在于,所述卷积层的卷积核大小为3乘3;所述conv层的卷积核大小为5乘5。6.根据权利要求1至5任一项所述的方法,其特征在于,所述方法还...
【专利技术属性】
技术研发人员:许淳,江瑾,朱艺伟,杨显志,徐键,谢尧,
申请(专利权)人:南方电网数字电网研究院有限公司,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。