Abstract:Surface defect detection is crucial in industrial production, and due to the conveyor speed, real-time detection requires 30 to 60 Frames Per Second, which exceeds the capability of most existing methods. This demand for high FPS has driven the need for lightweight detection models. Despite significant advancements in deep learning-based detection that have enabled single-stage models such as the YOLO series to achieve relatively fast detection, existing methods still face challenges in detecting multi-scale defects and tiny defects on complex surfaces while maintaining detection speed. This study proposes a lightweight single-stage detection model called Lightweight Industrial Defect Detection Network with improved YOLO architecture for high-precision and real-time industrial defect detection. Firstly, we propose the Large Separable Kernel Spatial Pyramid Pooling module, which is a spatial pyramid pooling structure with a separable large kernel attention mechanism, significantly improving the detection rate of multi-scale defects and enhancing the detection rate of small target defects. Secondly, we improved the Backbone and Neck structure of YOLOv8n with Dual convolutional kernel Convolution and enhanced the faster implementation of Cross Stage Partial Bottleneck with 2 Convolutions (C2f) module in the Neck structure with Ghost Convolution and Decoupled Fully Connected (DFC) attention, reducing the computational and parameter overhead of the model while ensuring detection accuracy. Experimental results on the NEU-DET steel defect datasets and PCB defect datasets demonstrate that compared to YOLOv8n, LIDD-YOLO improves the recognition rate of multi-scale defects and small target defects while meeting lightweight requirements. LIDD-YOLO achieves a 3.2% increase in mean Average Precision (mAP) on the NEU-DET steel defect dataset, reaching 79.5%, and a 2.6% increase in mAP on the small target PCB defect dataset, reaching 93.3%. Moreover, it reduces the parameter count by 20.0% and Floating Point Operations by 15.5%, further meeting the requirements for lightweight and high-precision industrial defect detection models.

LF-YOLO: A Lighter and Faster YOLO for Weld Defect Detection of X-Ray Image

WD-YOLO: A More Accurate YOLO for Defect Detection in Weld X-ray Images

Research on Improved Lightweight Weld Defect Identification Algorithm Based on YOLOv5

Data enhanced YOLOv8s algorithm for X-ray weld defect detection

LIDD-YOLO: A Lightweight Industrial Defect Detection Network

An Improved YOLOX Model for Detecting Strip Surface Defects

A Novel Weld-Seam Defect Detection Algorithm Based on the S-YOLO Model

YOLO-Weld: A Modified YOLOv5-Based Weld Feature Detection Network for Extreme Weld Noise

CRGF-YOLO: An Optimized Multi-Scale Feature Fusion Model Based on YOLOv5 for Detection of Steel Surface Defects

AMFF-YOLOX: Towards an Attention Mechanism and Multiple Feature Fusion Based on YOLOX for Industrial Defect Detection

A Lightweight Model of Automatic Pixel-Level Detection for Weld Defects

High-accuracy and Lightweight Weld Surface Defect Detector Based on Graph Convolution Decoupling Head

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

Online Pipeline Weld Defect Detection for Magnetic Flux Leakage Inspection System Via Lightweight Rotated Network

YOLO-LFPD: A Lightweight Method for Strip Surface Defect Detection

Generalized Weld Bead Region of Interest Localization and Improved Faster R-CNN for Weld Defect Recognition

A weld feature points detection method based on improved YOLO for welding robots in strong noise environment

Research on surface defect detection algorithm of pipeline weld based on YOLOv7

Convolutional Neural Network Defect Detection Algorithm for Wire Bonding X-ray Images

Yolo-MSAPF: Multiscale Alignment Fusion With Parallel Feature Filtering Model for High Accuracy Weld Defect Detection

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