arrow
Return

DHNet: a surface defect detection model utilizing multi-scale convolutional kernels

delete2025-01-17
delete1
PRE
AI
张莹莹 (Yingying Zhang) *
王硕 (Shuo Wang)
J
Jinhai Wang
Y
Yu Zhao
陈志为 cover
陈志为 (Zhiwei Chen)
DOI:10.1007/s11554-025-01623-zdelete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Detecting micro-defects in densely populated printed circuit boards (PCBs) with complex backgrounds is a critical challenge. To address the problem, the DHNet, a small object detection network based on YOLOv8 employing multi-scale convolutional kernels is proposed for feature extraction and fusion. The lightweight VOVGSHet module is designed for feature fusion and a pyramid structure to efficiently leverage feature map relationships while minimizing model complexity and parameters. Otherwise, to optimize the original extraction structure and enhance multi-scale defect detection, convolutional kernels of varying sizes process the same input channels. Additionally, the incorporation of the Wise-IoU loss function improves small defect detection accuracy and efficiency. Moreover, extensive experiments on a custom PCB dataset demonstrate DHNet's effectiveness, achieving an outstanding mean Average Precision (mAP) of 84.5%, surpassing the original YOLOv8 network by 4.0%, with parameters only of 2.85 M. Model demonstrates a latency of 3.6 ms on NVIDIA 4090. However, YOLOv8n has a latency of 4.4 ms. Validation on public DeepPCB and NEU datasets further confirms DHNet's superiority, which can reach 99.1% and 79.9% mAP, respectively. Finally, successful deployment on the NVIDIA Jetson Nano platform validates DHNet's suitability for real-time defect detection in industrial applications.
Keywords:
Printed circuit boards (PCBs)
Defect detection
Deep learning
Multi-scale feature fusion
Jetson Nano

Journal

Journal of Real-Time Image Processing cover
Journal of Real-Time Image Processing
IF:
3
Papers:
371
Citations:
2.0K

Organization

H
hefei univ technol
Scholars:
1.9K
Papers: 759
Citations: 248