arrow
返回

Lightweight PCB defect detection algorithm based on MSD-YOLO

delete2023-10-20
delete10
delete
OA
AI
G
Guoao Zhou
L
Lijuan Yu
苏义鑫 封面图
苏义鑫 (Yixin Su) *
徐
徐冰瑢 (Bingrong Xu)
G
Guoyuan Zhou
DOI:10.1007/s10586-023-04156-xdelete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Aiming at the problems of low accuracy and slow detection rate of existing target detection algorithms for PCB defect detection, and too many parameters of algorithm model leading to the inability to deploy on mobile terminals, a PCB defect detection algorithm based on MSD-YOLOv5 is proposed. Firstly, to ensure both detection accuracy and speed while reducing the model's size,we combine the lightweight MobileNet-v3 network with the CSPDarknet53 network. Further,the attention mechanism is introduced to highlight the important feature channels and weaken the less useful ones, so as to improve the feature extraction ability of the network. Finally, the coupling detection head is replaced with a decoupling detection head, and the defect location information and category information on the PCB are extracted and learned respectively, which solves the problem of highly coupling of different information feature distributions and enhances the generalization ability of the model. We conduct experiments on a publicly available PCB defect dataset from Peking University using this algorithm. The results show that the proposed method reduces the parameters of YOLOv5 model by 46% and improves the detection accuracy by 3.34%.
Keyword:
PCB
Defect recognition
YOLOv5
Lightweight network

期刊

C
Cluster Computing-The Journal of Networks Software Tools and Applications
IF:
4.1
论文数:
5.1K
被引数:
7.5K

机构

H
Huazhong Agricultural University
学者数:
3.2W
论文数: 1.8W
被引数: 3.5W
W
Wuhan University of Technology
学者数:
3.4W
论文数: 2.4W
被引数: 4.4W
引用论文

引用论文

err分享
err收藏
Identifying crop diseases using attention embedded MobileNet-V2 model
err2021-12-01
err51
PREAI
errChen, Junde; Zhang, Defu; Suzauddola, Md; Zeb, Adnan
err分享
err收藏
学者 查看更多内容