基于ECA-UNet的GH4169高温合金组织缺陷检测

摘要：

针对传统分割算法对于金属组织缺陷分割不连续,易被伪缺陷干扰的问题,提出一种结合迁移学习和通道注意力的ECA-UNet网络结构,实现GH4169高温合金组织缺陷图像分割与检测.首先,使用迁移学习将在ImageNet预训练好的参数迁移到ECA-UNet网络中,然后引用DenseNet网络思想将每一个有效特征层提取的图像信息都密集连接到下一个有效特征层中,并结合通道注意力模块组成加强跳跃连接模块,使UNet获得更准确的分割结果;最后,采用混合损失函数替代单一损失函数解决GH4169高温合金缺陷数据集正负样本不均衡的问题.与现行优秀算法PSPNet、Deeplabv3+缺陷分割效果对比发现ECA-UNet相较其他方法最佳指标值组合依然有2.91％,0.33％,3.5％的提升.结果表明:采用改进UNet分割缺陷的3个指标数据IoU,PA,F1-score分别为76.43％,87.22％,90.3％,相较于原始UNet网络3个指标分别提升了9.83％,9.09％,7.3％.

关键词：缺陷分割、通道注意力模块、迁移学习、加强跳跃连接模块、损失函数

所属期刊栏目：43

分类号：TP391.4(计算技术、计算机技术)

资助基金：国家自然科学基金;工业和信息化部绿色制造系统集成项目;河北省重点研发计划项目

在线出版日期：2023-03-07（万方平台首次上网日期，不代表论文的发表时间）

页数：共8页

页码：201-208