PCB Defect Detection Algorithm Based On YT-YOLO

被引：1

作者：

TangJian ^{[1
]}

YangYang ^{[1
]}

HouBaoshuai ^{[1
]}

HaoChongqing ^{[1
]}

机构：

[1] Hebei Univ Sci & Technol, Sch Elect Engn, Shijiazhuang, Hebei, Peoples R China

来源：

2023 35TH CHINESE CONTROL AND DECISION CONFERENCE, CCDC | 2023年

关键词：

YT-YOLO; SRGAN; YT-Block; Lightweight; Automatic;

D O I：

10.1109/CCDC58219.2023.10326719

中图分类号：

TP [自动化技术、计算机技术];

学科分类号：

0812 ;

摘要：

The quality of printed circuit board (PCB) has an important impact on electronic products. Aiming at the defects of PCB, this paper proposes a lightweight detection algorithm model YT-YOLO. Part of the dataset consists of PCB defect data publicly released by Peking University laboratory. SRGAN and data augmentation are used to increase the sample feature granularity and eliminate background noise, respectively. The designed YT Block is used to replace the original architecture to strengthen the feature Extraction ability. Compared with the original model, the parameters are reduced by 16.5%, the prediction accuracy is achieved by 93.5%, and the detection speed is improved by 13.4%. It can be directly deployed in the application terminal with limited computational power. It makes it possible to replace manual quality inspection with automatic, efficient and accurate inspection in the whole process.

引用

页码：976 / 981

页数：6

共 11 条

[1] Learning a Deep Convolutional Network for Image Super-Resolution [J].

Dong, Chao ;

Loy, Chen Change ;

He, Kaiming ;

Tang, Xiaoou .

COMPUTER VISION - ECCV 2014, PT IV, 2014, 8692 :184-199

[2]

Dosovitskiy A., 2020, ICLR 2021

[3] Rich feature hierarchies for accurate object detection and semantic segmentation [J].

Girshick, Ross ;

Donahue, Jeff ;

Darrell, Trevor ;

Malik, Jitendra .

2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, :580-587

[4] Deep Residual Learning for Image Recognition [J].

He, Kaiming ;

Zhang, Xiangyu ;

Ren, Shaoqing ;

Sun, Jian .

2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778

[5] Searching for MobileNetV3 [J].

Howard, Andrew ;

Sandler, Mark ;

Chu, Grace ;

Chen, Liang-Chieh ;

Chen, Bo ;

Tan, Mingxing ;

Wang, Weijun ;

Zhu, Yukun ;

Pang, Ruoming ;

Vasudevan, Vijay ;

Le, Quoc V. ;

Adam, Hartwig .

2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :1314-1324

[6]

Hu J, 2018, PROC CVPR IEEE, P7132, DOI [10.1109/TPAMI.2019.2913372, 10.1109/CVPR.2018.00745]

[7] ImageNet Classification with Deep Convolutional Neural Networks [J].

Krizhevsky, Alex ;

Sutskever, Ilya ;

Hinton, Geoffrey E. .

COMMUNICATIONS OF THE ACM, 2017, 60 (06) :84-90

[8]

Redmon J, 2018, Arxiv, DOI [arXiv:1804.02767, 10.48550/arXiv.1804.02767]

[9]

Simonyan K, 2015, Arxiv, DOI arXiv:1409.1556

[10]

Szegedy C, 2014, Arxiv, DOI arXiv:1312.6199

← 1 2 →