Abstract:To improve the detection ability for small defects on the surface of the metal base of an infrared laser sensor, master the fluctuation and distribution of product quality, and form closed-loop control of production and quality improvement, the advanced You Only Look Once (YOLO) v5s (an improved YOLO model) object detection algorithm was further improved in this study. Specifically, the same-scale feature fusion part that is easily ignored in the structure was strengthened to enhance the network detection performance. In deep learning object detection, the propagation of information between features in the neural network is important. It was often represented by the pyramid features in the neck part of the model to enhance feature fusion. First, this study proposed a cross-convolution feature strengthening connection method combining the backbone and neck, which shortened the path of information propagation and improved the semantic information between feature pyramids. Then, the concat module of the original network was improved, and a new enhanced feature concat module was proposed to enhance the fusion of features at the same scale. The attention modules implemented by combining the convolutional block attention module were integrated into the concat module to enable the network to learn the weights of each channel independently, enhance the information transmission between features, and improve the detection performance of deep learning small objects. Lastly, the K-means + + algorithm was used to optimize the self-made Metal Base dataset of Infrared Laser Sensor (ILS-MB) and generate a new anchor box suitable for small objects in this dataset to improve the matching degree of target objects. With a small increase in computational cost, the improved YOLO v5s algorithm enhanced the accuracy by 3.8% on the ILS-MB dataset and achieved a very significant effect compared with other state-of-the-art detection methods.

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

Small-scale defect detection in industrial environment based on lightweight deep learning network

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

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

Yolo-sd: simulated feature fusion for few-shot industrial defect detection based on YOLOv8 and stable diffusion

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

Surface Defect Detection Algorithm Based on Feature-Enhanced YOLO

Surface defect detection of industrial components based on vision

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

Integrated Circuit Packaging Defect Analysis and Deep Learning Detection Method

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

YOLO-RFF: An Industrial Defect Detection Method Based on Expanded Field of Feeling and Feature Fusion

Small object detection using deep feature learning and feature fusion network

Surface Defect Detection of Industrial Parts Based on YOLOv5

Enhanced feature Fusion structure of YOLO v5 for detecting small defects on metal surfaces

Improved YOLOv7-based steel surface defect detection algorithm

Metal Surface Defect Detection Using Modified YOLO

PCB Surface Defect Fast Detection Method Based on Attention and Multi-Source Fusion.

A lightweight and efficient model for surface tiny defect detection

Metal sensor base defects detection using deep learning based YOLO network

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