Abstract:Industrial defect detection is of great significance to ensure the quality of industrial products. The surface defects of industrial products are characterized by multiple scales, multiple types, abundant small objects, and complex background interference. In particular, small object detection of multiscale defects under complex background interference poses significant challenges for defect detection tasks. How to improve the algorithm's ability to detect industrial defects, especially in enhancing the detection capabilities of small-sized defects, while ensuring that the inference speed is not overly affected is a long-term prominent challenge. Aiming at achieving accurate and fast detection of industrial defects, this paper proposes an anchor-free network with DsPAN for small object detection. The core of this method is to propose a new idea i.e., feature enhancement in the feature fusion network for the feature information of small-size objects. Firstly, anchor-free YOLOv8 is adopted as the basic framework for detection to eliminate the affections of hyperparameters related to anchors, as well as to improve the detection capability of multi-scale and small-size defects. Secondly, considering the PAN path (top layer of neural networks for feature fusion) is more task-specific, we advocate that the underlying feature map of the PAN path is more vulnerable to small object detection. Hence, we in-depth study the PAN path and point out that the standard PAN will encounter several drawbacks caused by losing local details and position information in deep layer. As an alternative, we propose a lightweight and detail-sensitive PAN (DsPAN) for small object detection of multiscale defects by designing an attention mechanism embedded feature transformation module(LCBHAM) and optimizing the lightweight implementation. Our proposed DsPAN is very easy to be incorporated in YOLO series framework. The proposed method is evaluated on three public datasets, NEU-DET, PCB-DET, and GC10-DET. The mAP of the method is 80.4%, 95.8%, and 76.3%, which are 3.6%, 2.1%, and 3.9% higher than that of YOLOv8 and significantly higher than the state-of-the-art (SOTA) detection methods. Also, the method achieves the second-highest inference speed among the thirteen models tested. The results indicate that DsP-YOLO is expected to be used for real-time defect detection in industry.

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

Real-Time Defect Detection Model in Industrial Environment Based on Lightweight Deep Learning Network

LIDD-YOLO: A Lightweight Industrial Defect Detection Network

Surface Defect Inspection Under A Small Training Set Condition

Segmentation and Detection of Industrial Surface Defect Based on Deep Neural Network

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

Multi-scale Attention and Dilation Network for Small Defect Detection

A lightweight and efficient model for surface tiny defect detection

An efficient lightweight convolutional neural network for industrial surface defect detection

A One‐Stage Deep Learning Model for Industrial Defect Detection

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

A Lightweight Deep Network for Defect Detection of Insert Molding Based on X-ray Imaging

Surface defect detection of industrial components based on vision

DsP-YOLO: An anchor-free network with DsPAN for small object detection of multiscale defects

LE–MSFE–DDNet: a defect detection network based on low-light enhancement and multi-scale feature extraction

A Two-Stage Industrial Defect Detection Framework Based on Improved-YOLOv5 and Optimized-Inception-ResnetV2 Models

An Algorithm for Real-Time Aluminum Profile Surface Defects Detection Based on Lightweight Network Structure

KD-LightNet: A Lightweight Network Based on Knowledge Distillation for Industrial Defect Detection

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

Surface Defect Detection of Industrial Parts Based on YOLOv5

A Lightweight Strip Steel Surface Defect Detection Network Based on Improved YOLOv8