Fast PCB Defect Detection Method Based on FasterNet Backbone Network and CBAM Attention Mechanism Integrated With Feature Fusion Module in Improved YOLOv7

被引:30
作者
Chen, Boyuan [1 ,2 ]
Dang, Zichen [3 ]
机构
[1] Xi An Jiao Tong Univ, Sch Elect Engn, Xian 710049, Peoples R China
[2] Paio Tech Consulting Serv Co Ltd, Taicang 215400, Peoples R China
[3] Nanyang Technol Univ, Sch Elect & Elect Engn, Singapore 639798, Singapore
关键词
Feature extraction; Convolution; Computational modeling; Testing; Task analysis; Measurement by laser beam; Inspection; Printed circuits; Printed circuit board; FasterNet; attention module; YOLOv7;
D O I
10.1109/ACCESS.2023.3311260
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Printed Circuit Board (PCB) is a widely used electronic component and plays a critical role in the miniaturization and integration of circuits. However, the detection of PCB defects based on deep learning still encounter difficulties of limited efficiency. In order to address the issues of low speed and accuracy in PCB defect detection process, this paper proposed an innovative PCB defect detection method based on YOLOv7. Firstly, FasterNet was applied as the backbone network structure. With the new partial convolution, the spatial features were extracted more efficiently and the detection speed was improved by reducing redundant computations. Secondly, CBAM attention mechanism was integrated with feature fusion module, which allowed the model to selectively attend to relevant feature channels and spatial locations, thereby enhancing the discriminative ability of the feature representation and improving the accuracy. The experimental results indicated that the proposed model was superior to the traditional network on both PCB defect detection speed and accuracy. (1) The detection speed was increased from 54.3 frames per second to 83.3 frames per second. (2) The mAP0.5 reached 97.5% and mAP0.5:0.95 was increased from 52% to 54.7%. These improvements in speed and accuracy made it a more efficient solution for PCB defect detection.
引用
收藏
页码:95092 / 95103
页数:12
相关论文
共 28 条
[1]  
Bochkovskiy A, 2020, Arxiv, DOI [arXiv:2004.10934, 10.48550/arXiv.2004.10934, DOI 10.48550/ARXIV.2004.10934]
[2]   Run, Don't Walk: Chasing Higher FLOPS for Faster Neural Networks [J].
Chen, Jierun ;
Kao, Shiu-Hong ;
He, Hao ;
Zhuo, Weipeng ;
Wen, Song ;
Lee, Chul-Ho ;
Chan, S. -H. Gary .
2023 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2023, :12021-12031
[3]   PCB Defect Detection Method Based on Transformer-YOLO [J].
Chen, Wei ;
Huang, Zhongtian ;
Mu, Qian ;
Sun, Yi .
IEEE ACCESS, 2022, 10 :129480-129489
[4]   TDD-net: a tiny defect detection network for printed circuit boards [J].
Ding, Runwei ;
Dai, Linhui ;
Li, Guangpeng ;
Liu, Hong .
CAAI TRANSACTIONS ON INTELLIGENCE TECHNOLOGY, 2019, 4 (02) :110-116
[5]   RepVGG: Making VGG-style ConvNets Great Again [J].
Ding, Xiaohan ;
Zhang, Xiangyu ;
Ma, Ningning ;
Han, Jungong ;
Ding, Guiguang ;
Sun, Jian .
2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, :13728-13737
[6]  
Howard AG, 2017, Arxiv, DOI arXiv:1704.04861
[7]   Fast R-CNN [J].
Girshick, Ross .
2015 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2015, :1440-1448
[8]   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
[9]   Adversarial autoencoder for detecting anomalies in soldered joints on printed circuit boards [J].
Goto, Keisuke ;
Kato, Kunihito ;
Saito, Takaho ;
Aizawa, Hiroaki .
JOURNAL OF ELECTRONIC IMAGING, 2020, 29 (04)
[10]   GhostNet: More Features from Cheap Operations [J].
Han, Kai ;
Wang, Yunhe ;
Tian, Qi ;
Guo, Jianyuan ;
Xu, Chunjing ;
Xu, Chang .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2020, :1577-1586