Defects on steel surfaces are critical to product quality and safety in industrial production. However, existing inspection models for industrial production lines often face challenges such as high computational resource consumption and suboptimal detection performance. To address these issues, this paper proposes an enhanced model FAP-YOLO, based on YOLOv8, to improve the recognition of complex defects. FAP-YOLO introduces a redesigned lightweight C2f module with the StarBlock (via Star Operation), reducing latency while maintaining efficiency. Building on this, we propose the lightweight fusion attention mechanism (LFAM). Feature extraction is significantly improved by integrating local, channel, and spatial attention mechanisms with optimized weighting strategies. Additionally, the PIoU v2 loss function is employed, integrating target-size adaptation with anchor-box quality optimization to accelerate model convergence. On the NEU-DET and GC10-DET datasets, FAP-YOLO achieves mAP scores of 80.6% and 68.3%, surpassing the baseline by 4.2% and 1.8%, respectively. These results validate FAP-YOLO’s effectiveness in steel surface defect detection.

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

Efficient Surface Defect Detection via Multi-scale Features and Lightweight Attention

  • Xianming Huang,
  • Hongxiang Yu,
  • Ainan Liang,
  • Yawen Gao,
  • Pengju Si

摘要

Defects on steel surfaces are critical to product quality and safety in industrial production. However, existing inspection models for industrial production lines often face challenges such as high computational resource consumption and suboptimal detection performance. To address these issues, this paper proposes an enhanced model FAP-YOLO, based on YOLOv8, to improve the recognition of complex defects. FAP-YOLO introduces a redesigned lightweight C2f module with the StarBlock (via Star Operation), reducing latency while maintaining efficiency. Building on this, we propose the lightweight fusion attention mechanism (LFAM). Feature extraction is significantly improved by integrating local, channel, and spatial attention mechanisms with optimized weighting strategies. Additionally, the PIoU v2 loss function is employed, integrating target-size adaptation with anchor-box quality optimization to accelerate model convergence. On the NEU-DET and GC10-DET datasets, FAP-YOLO achieves mAP scores of 80.6% and 68.3%, surpassing the baseline by 4.2% and 1.8%, respectively. These results validate FAP-YOLO’s effectiveness in steel surface defect detection.