Steel Surface Defect Detection Algorithm Based on Improved YOLOv11n
To address the issues of low small-object detection accuracy and high computational cost in existing steel surface defect algorithms, this paper proposes CAP-YOLOv11n, an enhanced model based on YOLOv11n. Static local and dynamic global aggregation matrices are introduced before the detection head to fuse contextual information, strengthening key feature representations and suppressing background noise. An adaptive threshold mechanism combined with dynamic loss adjustment mitigates positive-negative sample imbalance and improves smallobject detection. Additionally, partial convolution is incorporated into the C3K2 module, applying convolution selectively to key channels, thus reducing computational redundancy. Experiments show CAP-YOLOv11n achieves an average precision of 81.6 %, a <tex xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink">$\mathbf{5. 4 \%}$</tex> improvement over the baseline, with significantly lower computational costs.
Authors
- Haokui Xu (ORCID: https://orcid.org/0000-0002-5924-8903)
- Yu Liu (ORCID: https://orcid.org/0000-0002-5216-3181)
- Wentao Wang (ORCID: https://orcid.org/0000-0001-9919-7488)
- Linen Zou
Institutions
- Minzu University of China (CN)
Publication Details
- Published
- 2025-05-23
- DOI
- https://doi.org/10.1109/cvidl65390.2025.11086055
- Primary Topic
- Industrial Vision Systems and Defect Detection
- Type
- article
- Field-Weighted Citation Impact
- 0.00