Design of lightweight metal surface defect detection technology for YOLOv7-tiny using Anchor-Free algorithm
Yi-Cheng Huang, Jun-Chang Lin, Yi-Ze Wu
TL;DR
This paper introduces a lightweight, real-time metal surface defect detection system using a modified YOLOv7-tiny model for improved accuracy and speed in industrial applications.
Contribution
A novel anchor-free YOLOv7-tiny architecture with enhancements for real-time metal surface defect detection is proposed.
Findings
The model achieved a 6.24% increase in mean average precision (mAP) compared to existing methods.
The improved model operates at over 90 FPS, demonstrating strong real-time performance.
The architecture effectively handles defects with extreme aspect ratios and small-scale features.
Abstract
Accurate identification of defect morphology and size distribution on metallic surfaces is crucial for manufacturing, particularly in the steel industry where such materials are widely used. This study proposes a novel lightweight architecture based on YOLOv7-tiny to address the challenges of real-time metallic surface defect detection. The original anchor-based detection head is replaced with an anchor-free mechanism to reduce missed detections of defects with extreme aspect ratios, while a logarithmic transformation–based enhancement is introduced to strengthen defect features. The backbone is replaced by the lightweight MobileNetV3-large network to reduce the number of parameters, and an Efficient Multi-Scale Attention (EMA) module is embedded into the bottleneck structure to increase target importance and suppress background interference, especially for small defects. In addition, a…
Genes, proteins, chemicals, diseases, species, mutations and cell lines named across the full text — each resolved to its canonical identifier and authoritative record.
Click any figure to enlarge with its caption.
Figure 10
Figure 11
Figure 12
Figure 13
Figure 14
Figure 15
Figure 16
Figure 17
Figure 18
Figure 19
Figure 1
Figure 20
Figure 21
Figure 22
Figure 23
Figure 24
Figure 25
Figure 26
Figure 27
Figure 2
Figure 3
Figure 4
Figure 5
Figure 6
Figure 7
Figure 8
Figure 9Peer Reviews
No public reviews on file for this paper yet. If you reviewed it on a platform where reviews are public (OpenReview, ICLR, NeurIPS, ICML), you can paste yours below so the community can read it here.
Videos
No videos yet. Explain this paper in a talk, walkthrough, or lecture? Add one.
Taxonomy
TopicsAdvanced Neural Network Applications · Industrial Vision Systems and Defect Detection · Advanced Data and IoT Technologies