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.

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

Detection Algorithm for Bearing Roller End Surface Defects Based on Improved YOLOv5n and Image Fusion

Metal sensor base defects detection using deep learning based YOLO network

Metal Surface Defect Detection Using Modified YOLO

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

A Lightweight YOLO Object Detection Algorithm Based on Bidirectional Multi‐Scale Feature Enhancement

LSD-YOLOv5: A Steel Strip Surface Defect Detection Algorithm Based on Lightweight Network and Enhanced Feature Fusion Mode

Efficient detection of multiscale defects on metal surfaces with improved YOLOv5

Improved YOLOv7-based steel surface defect detection algorithm

CFM-YOLOv5:CFPNet moudle and muti-target prediction head incorporating YOLOv5 for metal surface defect detection

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

An improved YOLOv5 method for large objects detection with multi-scale feature cross-layer fusion network

MCF-YOLOv5: A Small Target Detection Algorithm Based on Multi-Scale Feature Fusion Improved YOLOv5

A stable diffusion enhanced YOLOV5 model for metal stamped part defect detection based on improved network structure

LCG-YOLO: A Real-Time Surface Defect Detection Method for Metal Components

Metal surface defect detection based on improved YOLOv5

MSFT-YOLO: Improved YOLOv5 Based on Transformer for Detecting Defects of Steel Surface

FE-YOLOv5: Feature enhancement network based on YOLOv5 for small object detection

Improving Object Detection in YOLOv8n with the C2f-f Module and Multi-Scale Fusion Reconstruction

Mixed Receptive Fields Augmented YOLO with Multi-Path Spatial Pyramid Pooling for Steel Surface Defect Detection