A lightweight image-level segmentation method for steel surface defects based on cross-layer feature fusion

Abstract

Steel is widely used in the aerospace, machinery and automotive industries. Surface defects not only have a negative impact on the appearance of steel but also significantly reduce its wear resistance, high temperature resistance, corrosion resistance and fatigue strength. Therefore, the detection of steel surface defects is very important to improve the quality of steel production. The limited availability of surface defect samples in the industrial sector poses significant challenges for the accurate detection of defects in high-quality materials. In addition, the existing defect detection model is highly complex and not easy to deploy. To solve this problem, a lightweight defect detection network suitable for steel defects is proposed. The cross-layer feature fusion (CFF) in the design enables effective utilisation of multi-layer semantic features, facilitating the detection of small defects in steel. Secondly, a new loss function is designed to make up for the problems of small data volume and uneven data distribution. The experimental results demonstrate that the steel surface defect detection method proposed in this paper achieves the highest detection performance on widely used public datasets such as RSDDS, NEUS and NRSD-CR(test), while maintaining the lowest model complexity.