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.

A more efficient method for wire rope surface defect detection based on fusing cross-stage features and lightweight convolution modules

TBi-YOLOv5: A Surface Defect Detection Model for Crane Wire with Bottleneck Transformer and Small Target Detection Layer

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

Non-destructive Testing of Wire Rope Algorithm Based on Lightweight YOLOv7-tiny

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

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

A Highly Efficient and Lightweight Detection Method for Steel Surface Defect

Enhanced Darknet53 Combine MLFPN Based Real-Time Defect Detection in Steel Surface

LIDD-YOLO: A Lightweight Industrial Defect Detection Network

Visual Sensing Inspection for the Surface Damage of Steel Wire Ropes With Object Detection Method

Deep Learning Based Defect Detection Method for Overhead Transmission Wires

SDD-YOLO: A Lightweight, High-Generalization Methodology for Real-Time Detection of Strip Surface Defects

YOLO-RDP: Lightweight Steel Defect Detection through Improved YOLOv7-Tiny and Model Pruning

Efficient Detection Model of Steel Strip Surface Defects Based on YOLO-V7

Strip steel surface defect detection based on lightweight YOLOv5

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

A modified-YOLOv5s model for detection of wire braided hose defects

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

An Improved Lightweight YOLOv5 Algorithm for Detecting Railway Catenary Hanging String

Surface Defect Detection for Automated Tape Laying and Winding Based on Improved YOLOv5

Steel Surface Defect Detection Method Based on Improved YOLOX