Abstract:The detection of surface defects on metal products during the production process is crucial for ensuring high-quality products. These defects also lead to significant losses in the high-tech industry. To address the issues of slow detection speed and low accuracy in traditional metal surface defect detection, an improved algorithm based on the YOLOv7-tiny model is proposed. Firstly, to enhance the feature extraction and fusion capabilities of the model, the depth aware convolution module (DAC) is introduced to replace all ELAN-T modules in the network. Secondly, the AWFP-Add module is added after the Concat module in the network's Head section to strengthen the network's ability to adaptively distinguish the importance of different features. Finally, in order to expedite model convergence and alleviate the problem of imbalanced positive and negative samples in the study, a new loss function called Focal-SIoU is used to replace the original model's CIoU loss function. To validate the effectiveness of the proposed model, two industrial metal surface defect datasets, GC10-DET and NEU-DET, were employed in our experiments. Experimental results demonstrate that the improved algorithm achieved detection frame rates exceeding 100 fps on both datasets. Furthermore, the enhanced model achieved an mAP of 81% on the GC10-DET dataset and 80.1% on the NEU-DET dataset. Compared to the original YOLOv7-tiny algorithm, this represents an increase in mAP of nearly 11% and 9.2%, respectively. Moreover, when compared to other novel algorithms, our improved model demonstrated enhanced detection accuracy and significantly improved detection speed. These results collectively indicate that our proposed enhanced model effectively fulfills the industry's demand for rapid and efficient detection and recognition of metal surface defects.

Surface Defect Detection Based on Improved YOLOv3-Tiny Algorithm

Surface Defect Detection with Modified Real-Time Detector YOLOv3

Object Detection Algorithm for Surface Defects Based on a Novel YOLOv3 Model

Surface Defect Detection Algorithm Based on Feature-Enhanced YOLO

A real-time and efficient surface defect detection method based on YOLOv4

A Small-scale Object Detection Method for LCD Defects Based on Improved YOLOv8

Attention-based deep learning for chip-surface-defect detection

A Novel YOLOv5_ES based on lightweight small object detection head for PCB surface defect detection

Small object detection method with shallow feature fusion network for chip surface defect detection

Improved YOLOv7-based steel surface defect detection algorithm

Surface defect detection of industrial components based on vision

Improving the Algorithm Study of YOLO in Steel Surface Defect Detection

NHD‐YOLO: Improved YOLOv8 using optimized neck and head for product surface defect detection with data augmentation

Defect Detection Algorithm for Steel Surface Based on Improved YOLOv5

ST-CA YOLOv5: Improved YOLOv5 Based on Swin Transformer and Coordinate Attention for Surface Defect Detection

Usage of an Improved YOLOv5 for Steel Surface Defect Detection

Unleashing the power of AI in detecting metal surface defects: an optimized YOLOv7-tiny model approach

Online Detection of Surface Defects Based on Improved YOLOV3

Toward surface defect detection in electronics manufacturing by an accurate and lightweight YOLO-style object detector

Surface defect detection of ceramic disc based on improved YOLOv5s

A method for workpiece surface small-defect detection based on CutMix and YOLOv3