Abstract:Surface defect detection in industrial environments is crucial for quality management and has significant research value. General detection networks, such as the YOLO series, have proven effective in various dataset detections. However, due to the complex and varied surface defects of industrial products, many defects occupy a small proportion of the surface and fall into the category of typical small target detection problems. Moreover, the complexity of general detection network architectures relies on high-tech hardware, making it difficult to deploy on devices without GPUs or on edge computing and mobile devices. To meet the practical needs of industrial product defect inspection applications, this paper proposes a lightweight network specifically designed for defect detection in industrial fields. This network is composed of four parts: a backbone network, a multiscale feature aggregation network, a residual enhancement network, and an attention enhancement network. The network includes a backbone network that integrates attention layers for feature extraction, a multiscale feature aggregation network for semantic information, a residual enhancement network for spatial focus, and an attention enhancement network for global–local feature interaction. These components enhance detection performance for diverse defects while maintaining low hardware requirements. Experimental results show that this network outperforms the latest and most popular YOLOv5n and YOLOv8n models in the five indicators P, R, F1, mAP@.5, and GFLOPS when used on four public datasets. It even approaches or surpasses the YOLOv8s and YOLOv5s models with several times the GFLOPS computation. It balances the requirements of lightweight real-time and accuracy in the scenario of industrial product surface defect detection.

Research on Quartz Wafer Defect Detection Method Based on Deep Learning Model Pruning

A Wafer Surface Defect Detection Method Built on Generic Object Detection Network

Optimized YOLOv8 based on SGW for surface defect detection of silicon wafer

A Light-Weighted CNN Model for Wafer Structural Defect Detection

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

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

Optimization of workpiece detection model based on channel pruning and fixed-point quantization

Research on the Construction of an Efficient and Lightweight Online Detection Method for Tiny Surface Defects through Model Compression and Knowledge Distillation

Machine Learning Based Wafer Defect Detection

Using Filter Pruning-Based Deep Learning Algorithm for the Real-Time Fruit Freshness Detection with Edge Processors

Real-Time Grading of Defect Apples Using Semantic Segmentation Combination with a Pruned YOLO V4 Network

Wafer Defect Localization and Classification Using Deep Learning Techniques

A lightweight and efficient model for surface tiny defect detection

Deep Learning-Based Detection of Defects in Wafer Buffer Zone During Semiconductor Packaging Process

Semiconductor Defect Detection by Hybrid Classical-Quantum Deep Learning

Collaborative Filter Pruning for Efficient Automatic Surface Defect Detection

Inspection and Classification of Semiconductor Wafer Surface Defects Using CNN Deep Learning Networks

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

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

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

Wafer Crack Detection Based on Yolov4 Target Detection Method