An Efficient and Accurate Quality Inspection Model for Steel Scraps Based on Dense Small-Target Detection

Abstract

Scrap steel serves as the primary alternative raw material to iron ore, exerting a significant impact on production costs for steel enterprises. With the annual growth in scrap resources, concerns regarding traditional manual inspection methods, including issues of fairness and safety, gain increasing prominence. Enhancing scrap inspection processes through digital technology is imperative. In response to these concerns, we developed CNIL-Net, a scrap-quality inspection network model based on object detection, and trained and validated it using images obtained during the scrap inspection process. Initially, we deployed a multi-camera integrated system at a steel plant for acquiring scrap images of diverse types, which were subsequently annotated and employed for constructing an enhanced scrap dataset. Then, we enhanced the YOLOv5 model to improve the detection of small-target scraps in inspection scenarios. This was achieved by adding a small-object detection layer (P2) and streamlining the model through the removal of detection layer P5, resulting in the development of a novel three-layer detection network structure termed the Improved Layer (IL) model. A Coordinate Attention mechanism was incorporated into the network to dynamically learn feature weights from various positions, thereby improving the discernment of scrap features. Substituting the traditional non-maximum suppression algorithm (NMS) with Soft-NMS enhanced detection accuracy in dense and overlapping scrap scenarios, thereby mitigating instances of missed detections. Finally, the model underwent training and validation utilizing the augmented dataset of scraps. Throughout this phase, assessments encompassed metrics like mAP, number of network layers, parameters, and inference duration. Experimental findings illustrate that the developed CNIL-Net scrap-quality inspection network model boosted the average precision across all categories from 88.8% to 96.5%. Compared to manual inspection, it demonstrates notable advantages in accuracy and detection speed, rendering it well suited for real-world deployment and addressing issues in scrap inspection like real-time processing and fairness.