TCS-Net：一种核电安全壳细小裂缝分割网络

doi:10.15961/j.jsuese.202100590

工程科学与技术 ›› 2022, Vol. 54 ›› Issue (05) : 249 -256. DOI: 10.15961/j.jsuese.202100590

作者信息 +

Author information +

文章历史 +

摘要

针对核电安全壳表面裂缝视觉检查任务面临的裂缝细小且占像素少、裂缝与背景对比度低、相似纹理干扰多、光照不均衡等问题，作者提出了一种细小裂缝分割模型（segmentation network of tiny cracks,TCS-Net）。该模型具有与U-Net网络相似的编码–解码结构，在下采样过程使用软池化（soft pooling）可实现在编码过程中保存原始图像基本特征的同时又保持了高响应特征信息的放大，从而减少信息损失实现裂缝边缘细节及位置信息保留；解码端在上采样过程中加入兼顾通道注意力和空间注意力的语义补偿模块以融合编码端的各层特征，可增强裂缝的多尺度细节信息；鉴于裂缝分割任务存在数据不平衡的分类问题，为避免训练过程被多数类（背景像素）所主导，TCS-Net模型结合二元交叉熵损失和Dice系数作为目标损失函数，以解决单一损失关注度倾向带来的训练不稳定的问题，也可优化准确率、交并比、召回率等性能指标。该模型在真实的安全壳图像上实验，结果表明，与现有的主流语义分割模型相比，TCS-Net裂缝分割模型的交并比指标可提高5%～9%，召回率指标可提高9%～13%，由此说明该模型具有检测率和检测精度更高，能有效适用于目标与背景严重不平衡、背景复杂且干扰较多情况下的细小裂缝分割任务。