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:
408
Citations:
2.0K

Organization

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

Cited Papers

Enhancing Geometric Factors in Model Learning and Inference for Object Detection and Instance Segmentation
err2022-08-01
err620
errOAAI
errZheng, Zhaohui; Wang, Ping; Ren, Dongwei; Liu, Wei; Ye, Rongguang; Hu, Qinghua; Zuo, Wangmeng
errShare
errSave
Adaptive EPI-Matching Cost for Light Field Disparity Estimation
err2024-01-01
err0
PREAI
errWang, Tun; Sheng, Hao; Chen, Rongshan; Cong, Ruixuan; Zhao, Mingyuan; Cui, Zhenglong
errShare
errSave
errShare
errSave
Martian dust devil statistics from high‐resolution large‐eddy simulations
err2016-05-14
err0
errOAAI
errSeiya Nishizawa; Masatsugu Odaka; Yoshiyuki O. Takahashi; Ko‐ichiro Sugiyama; Kensuke Nakajima; Masaki Ishiwatari; Shin‐ichi Takehiro; Hisashi Yashiro; Yousuke Sato; Hirofumi Tomita; Yoshi‐Yuki Hayashi
errShare
errSave
Review of vision-based defect detection research and its perspectives for printed circuit board
err2023-10-01
err26
PREAI
errZhou, Yongbing; Yuan, Minghao; Zhang, Jian; Ding, Guofu; Qin, Shengfeng
errShare
errSave
researcher View more