Optics and Precision Engineering, Volume. 32, Issue 7, 1075(2024)
Object detection of steel surface defect based on multi-scale enhanced feature fusion
To address the issue of low recognition accuracy in lightweight algorithms for steel surface defect detection, this paper introduces a Multi-scale Enhanced Feature Fusion (EFF) technique. Initially, an Adaptive Weighted Fusion (AWF) module calculates fusion weights adaptively for different feature levels. This allows shallow features to enrich with deep semantics without compromising detail. Subsequently, the Spatial Feature Enhancement (SFE) module boosts the fused features from three distinct directions and improves network stability by integrating residual pathways, enabling the convolution process to extract more critical information. The model then selects better training samples based on the overlap between the prior box and the ground truth. Experimental outcomes show that the proposed method achieves a detection accuracy of 80.47%, marking a 6.81% increase over the baseline algorithm. Moreover, with 2.36 M parameters and 952.67 MFLOPs, this algorithm efficiently and accurately identifies steel surface defects, demonstrating significant practical utility.
Get Citation
Copy Citation Text
Shanling LIN, Xueling PENG, Dong WANG, Zhixian LIN, Jianpu LIN, Tailiang GUO. Object detection of steel surface defect based on multi-scale enhanced feature fusion[J]. Optics and Precision Engineering, 2024, 32(7): 1075
Category:
Received: Oct. 24, 2023
Accepted: --
Published Online: May. 28, 2024
The Author Email: LIN Jianpu (ljp@fzu.edu.cn)