Automatically surface defect detection plays a crucial role during the industrial production process. Unfortunately, some special defects, such as weak and strip-like defects, are relatively difficult to classify and localize accurately. In this paper, we propose a novel Location-Aware Guidance Network for weak and strip defect detection, termed as LAGNet. To enhance the feature representation of weak defects, we introduce the Location Activation Map (LAM) by visualizing the confidence score map that indicates the probability of the existence of an object in each region. The LAM and RGB images are fed into the network in a parallel manner for feature extraction, and then we fuse these two branches via a Location Guidance Block (LGB) that inherently encodes comprehensive and complementary information for detection. Additionally, a Strip Convolution Enhancement Module (SCEM) is presented using the depthwise strip convolutions with long but narrow kernels and attention mechanism and plugged into the detection neck to model the long-range dependencies along both horizontal and vertical spatial directions, thus improving the detection performance of anisotropic defects with banded structures. Notably, LAGNet achieves the top-ranking results on two popular steel benchmarks and significantly outperforms the baseline network YOLOv5: 85.5% mAP (vs. 76.2%) on NEU-DET and 76.6% mAP (vs. 66.0%) on GC10-DET.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

LAGNet: A Location-Aware Guidance Network for Weak and Strip Defect Detection

  • Lisha Cui,
  • Helong Jiao,
  • Tengyue Liu,
  • Chunyan Niu,
  • Ming Ma,
  • Xiaoheng Jiang,
  • Mingliang Xu

摘要

Automatically surface defect detection plays a crucial role during the industrial production process. Unfortunately, some special defects, such as weak and strip-like defects, are relatively difficult to classify and localize accurately. In this paper, we propose a novel Location-Aware Guidance Network for weak and strip defect detection, termed as LAGNet. To enhance the feature representation of weak defects, we introduce the Location Activation Map (LAM) by visualizing the confidence score map that indicates the probability of the existence of an object in each region. The LAM and RGB images are fed into the network in a parallel manner for feature extraction, and then we fuse these two branches via a Location Guidance Block (LGB) that inherently encodes comprehensive and complementary information for detection. Additionally, a Strip Convolution Enhancement Module (SCEM) is presented using the depthwise strip convolutions with long but narrow kernels and attention mechanism and plugged into the detection neck to model the long-range dependencies along both horizontal and vertical spatial directions, thus improving the detection performance of anisotropic defects with banded structures. Notably, LAGNet achieves the top-ranking results on two popular steel benchmarks and significantly outperforms the baseline network YOLOv5: 85.5% mAP (vs. 76.2%) on NEU-DET and 76.6% mAP (vs. 66.0%) on GC10-DET.