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.

Surface Defect Detection and Semantic Segmentation with a Novel Lightweight Deep Neural Network

MFNet: A Novel Multilevel Feature Fusion Network With Multibranch Structure for Surface Defect Detection

A Sub-Region Unet for Weak Defects Segmentation with Global Information and Mask-Aware Loss.

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

A Deep Learning-Based Surface Defect Inspection System Using Multiscale and Channel-Compressed Features

AnomalySeg: Deep Learning-Based Fast Anomaly Segmentation Approach for Surface Defect Detection

A Lightweight Spatial and Temporal Multi-Feature Fusion Network for Defect Detection

AFFNet: An Attention-Based Feature-Fused Network for Surface Defect Segmentation

Detail-semantic guide network based on spatial attention for surface defect detection with fewer samples

Automated Detection of Defects with Low Semantic Information in X-ray Images Based on Deep Learning

An Adaptive Image Segmentation Network for Surface Defect Detection

Joining Spatial Deformable Convolution and a Dense Feature Pyramid for Surface Defect Detection

Attention-Based Multiscale Feature Fusion for Efficient Surface Defect Detection

DDSNet: Deep Dual-Branch Networks for Surface Defect Segmentation

A Pixel-Level Segmentation Convolutional Neural Network Based on Global and Local Feature Fusion for Surface Defect Detection

A lightweight and efficient model for surface tiny defect detection

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

An Efficient End-to-End Multitask Network Architecture for Defect Inspection

Decision Fusion Network with Perception Fine-tuning for Defect Classification

Unsupervised Defect Segmentation Via Forgetting-Inputting-Based Feature Fusion and Multiple Hierarchical Feature Difference

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