返回
Lightweight PCB defect detection algorithm based on MSD-YOLO
DOI:10.1007/s10586-023-04156-x.png)
摘要
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
IF:
4.1
论文数:
5.1K
被引数:
7.5K
机构
引用论文
Impact of climate variability on human health: A pilot study in tertiary care hospital of Eastern Uttar Pradesh, India
MAUSAM
IF0
A classification method for brain MRI via MobileNet and feedforward network with random weights基于MobileNet和随机权前馈网络的脑MRI分类方法
Detection of PCB Surface Defects With Improved Faster-RCNN and Feature Pyramid Network基于改进的Faster-RCNN和特征金字塔网络的PCB表面缺陷检测
IEEE ACCESS
IF3.6
MMF: A Multi-scale MobileNet based fusion method for infrared and visible imageMMF: 一种基于多尺度MobileNet的红外与可见光图像融合方法
Quantization Friendly MobileNet (QF-MobileNet) Architecture for Vision Based Applications on Embedded Platforms量化友好的MobileNet (qf-mobilenet) 架构,用于嵌入式平台上基于视觉的应用程序
NEURAL NETWORKS
IF6.3

