改进DRAEM的非监督纤维绳索表面缺陷检测算法
DOI:
CSTR:
作者:
作者单位:

河北工业大学机械工程学院天津300401

作者简介:

通讯作者:

中图分类号:

TP391.7

基金项目:

国家自然科学基金联合基金项目(U23A6017)、石家庄市科技项目(SJZZXA24005,SJZZXC24002)、石家庄科技局项目(SJZZXC23002)资助


Improved unsupervised DRAEM algorithm for surface defect detection in fiber ropes
Author:
Affiliation:

School of Mechanical Engineering, Hebei University of Technology, Tianjin 300401, China

Fund Project:

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

    针对细小纹理以及背景复杂的场景中纤维绳索表面缺陷检测存在小目标漏检率高、检测精度低和算法鲁棒性不足等问题,提出了一种基于判别式训练重建嵌入模型(discriminatively trained reconstruction embedding for surface anomaly detection, DRAEM)改进的纤维绳索表面缺陷算法。在预处理部分加入GrabCut分割算法提取每张图像的掩码,通过掩码约束异常生成减少背景影响,以避免复杂背景造成的误检的问题;在重构网络中加入跳跃连接来捕获高维图像空间正态数据分布,并添加通道和空间双重注意力模块以强调对异常区域重建的能力,以提高细小纹理的重构效果避免纹理丢失从而导致小尺度缺陷漏检的问题;在分割网络编码器后两层加入Transform模块优化对正常与异常的级联的全局特征的捕捉;同时用空洞空间金字塔池化(atrous spatial pyramid pool, ASPP),捕获不同尺度的上下文信息进行全局特征聚合,为分割提供足够的语义差异,以提高模型的分割精度和表面小目标缺陷的检测精度。实验结果表明,与原DRAEM对比图像级AUROC提高了4.4%,像素级AUROC提高了4.43%,像素级平均精度提高了21.86%,提高了模型的识别精度与鲁棒性,更好地应用于纤维绳索的缺陷检测。

    Abstract:

    To address the issues of high missed detection rates for small targets, low detection accuracy, and insufficient algorithm robustness in fiber rope surface defect detection under scenarios with fine textures and complex backgrounds, this paper proposes an improved fiber rope surface defect detection algorithm based on the unsupervised model a discriminatively trained reconstruction embedding for surface anomaly detection (DRAEM). In the preprocessing stage, the GrabCut segmentation algorithm is incorporated to extract masks for each image, reducing background interference through mask-constrained anomaly generation and mitigating false positives and missed detections caused by complex backgrounds. In the reconstruction network, skip connections are introduced to capture high-dimensional image space normal data distributions, and a dual channel-spatial attention module is added to enhance the reconstruction capability for anomalous regions, to improve the reconstruction quality of fine-scale textures, thereby avoiding texture loss and preventing the missed detection of small-scale defects. In the segmentation network, Transformer modules are integrated after the last two layers of the encoder to optimize the capture of cascaded global features between normal and abnormal regions. Additionally, atrous spatial pyramid pooling (ASPP) is employed to capture multi-scale contextual information for global feature aggregation, providing sufficient semantic differentiation for segmentation and enhancing the model’s segmentation accuracy and the detection accuracy of small surface defects. Experimental results demonstrate that, compared to the original DRAEM, the proposed method achieves a 4.4% improvement in image-level AUROC, a 4.43% increase in pixel-level AUROC, and a 21.86% boost in pixel-level average precision. These enhancements significantly improve the model’s recognition accuracy and robustness, making it more effective for fiber rope defect detection applications.

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

方忠志,朱华波,陶友瑞,石泽.改进DRAEM的非监督纤维绳索表面缺陷检测算法[J].电子测量与仪器学报,2026,40(2):44-54

复制
分享
相关视频

文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2026-04-30
  • 出版日期:
文章二维码
×
《电子测量与仪器学报》
关于防范虚假编辑部邮件的郑重公告