首页> 中文期刊> 《五邑大学学报:自然科学版》 >基于注意力密集连接Unet的磁瓦表面孔洞和裂痕缺陷分割算法

基于注意力密集连接Unet的磁瓦表面孔洞和裂痕缺陷分割算法

         

摘要

针对检测磁瓦孔洞和裂痕时容易丢失缺陷特征等问题,本文在Unet网络的基础上添加注意力模块CBAM来提取更受关注的特征,并在Unet网络的编码部分使用密集连接来充分利用特征.同时,为了减少Unet网络的过多池化下采样操作导致小物体特征消失,还使用膨胀率为[1,2,5,1,2,5]的混合空洞卷积取代池化操作.最后设置加权交叉熵损失函数解决磁瓦数据集当中样本分布不均的问题.实验证明,本文算法在磁瓦孔洞和裂痕缺陷分割时,MIoU分别提高了2.696%和2.739%,Dice系数分别提高了3.342%和2.602%.本文算法在一定程度上提升了磁瓦缺陷分割精确度,还改善了Unet网络边界分割模糊等问题.

著录项

相似文献

  • 中文文献
  • 外文文献
  • 专利
获取原文

客服邮箱:kefu@zhangqiaokeyan.com

京公网安备:11010802029741号 ICP备案号:京ICP备15016152号-6 六维联合信息科技 (北京) 有限公司©版权所有
  • 客服微信

  • 服务号