基于轻量化卷积块注意力Transformer的表面缺陷检测算法
CSTR:
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

TP391.4

基金项目:

国家自然科学基金青年科学基金(61702340)


Surface defect detection algorithm via lightweight convolutional block attention Transformer
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对缺陷语义分割任务中缺陷与背景差异小以及同类缺陷间差异大的问题,本文提出一种基于轻量化卷积块注意力Transformer的表面缺陷检测算法(LCBAFormer),旨在提高不同类型缺陷的分割准确率.首先,设计轻量化卷积块注意力模块LCBAM,通过结合通道注意力模块和空间注意力模块,提取有效的通道信息和空间信息,使模型更专注于局部缺陷特征信息,增强不同缺陷间的特征差异,减少同类缺陷间的差异.其次,提出轻量化的语义注入模块SIM,以逐步融合多尺度特征信息,提升网络对不同缺陷的定位和区分能力.在NEU-Seg钢带缺陷数据集和MT-Defect磁瓦缺陷数据集上的实验结果表明:本文提出的算法平均交并比(mIoU)分别为84.75%和79.46%;平均召回率(mRec)分别为92.29%和87.50%;平均F1分数(mF1)分别为91.52%和88.08%;算法的计算复杂度较低,每秒10亿次浮点运算次数(GFLOPs)分别为1.03和2.65.与主流算法相比,本文方法在参数量和检测性能上做到了较好的平衡,具有更小的参数量和更好的分割结果.

    Abstract:

    To address the issues of small differences between defects and background,as well as large variations within the same class of defects in defect semantic segmentation tasks,this paper proposes a surface defect detection algorithm based on Lightweight Convolutional Block Attention Transformer,named LCBAFormer.The proposed algorithm aims to enhance the accuracy of segmentation for various types of defects.Firstly,a Lightweight Convolutional Block Attention Module (LCBAM) is designed,which integrates channel attention and spatial attention modules to extract effective channel and spatial information.This enables the model to focus more on local defect feature information,enhance feature differences between defects,and mitigate variations within the same category of defects.Secondly,a lightweight Semantic Injection Module (SIM) is introduced,which gradually fuses multi-scale feature information and thereby improving the network's ability to locate and distinguish different defects.The experimental results show that on the NEU-Seg steel strip defect dataset and the magnetic tile defect dataset (MT-Defect),the proposed algorithm achieves mean Intersection over Union (mIoU) of 84.75% and 79.46%,mean Recall (mRec) rates of 92.29% and 87.50%,and mean F1 scores (mF1) of 91.52% and 88.08%,respectively.Additionally,the algorithm exhibits low computational complexity,with 1.03 and 2.65 GFLOPs (billion floating-point operations per second) on the NEU-Seg and MT-Defect datasets,respectively.Compared to mainstream methods,the proposed algorithm features fewer parameters and superior segmentation results,achieving a good balance between parameter count and detection performance.

    参考文献
    相似文献
    引证文献
引用本文

孙文赟,刘梓麟.基于轻量化卷积块注意力Transformer的表面缺陷检测算法[J].南京信息工程大学学报(自然科学版),2025,17(5):624-633
SUN Wenyun, LIU Zilin. Surface defect detection algorithm via lightweight convolutional block attention Transformer[J]. Journal of Nanjing University of Information Science & Technology, 2025,17(5):624-633

复制
分享
相关视频

文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2024-06-05
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2025-10-18
  • 出版日期: 2025-09-28
文章二维码

地址:江苏省南京市宁六路219号    邮编:210044

联系电话:025-58731025    E-mail:nxdxb@nuist.edu.cn

南京信息工程大学学报 ® 2026 版权所有  技术支持:北京勤云科技发展有限公司