基于轻量级卷积神经网络的载波芯片缺陷检测

doi:10.3778/j.issn.1002-8331.2104-0299

摘要/Abstract

摘要： 载波芯片（chip on carrier，COC）是光发射次模块（transmitter optical subassembly，TOSA）的重要组成部分，被广泛应用于光通信领域，实现光电转换。针对载波芯片崩口、定位柱破损以及波导污渍三种不同类别缺陷的实时检测问题，提出了一种基于轻量级卷积神经网络的载波芯片缺陷检测算法YOLO-Efficientnet。为了减少网络参数，缩短检测时间，采用轻量级卷积神经网络Efficientnet作为主干网络对图像进行特征提取，在移动翻转瓶颈卷积（MBConv）的基础上，引入了压缩与激发网络（SENet）的注意力思想，在通道维度上引入注意力机制；为了解决下采样的过程中导致信息丢失的问题，引入空间金字塔池化（SPP）结构来增大图像的感受野，分离出更加显著的上下文特征。针对COC缺陷多尺度以及波导区域污渍小目标难以检测的问题，引入了PANet结构进行多尺度特征融合。实验结果表明，提出的算法对COC缺陷检测的准确率达到了98.5%，检测时间达到每张图片0.42?s，满足实时检测的需求。

关键词: 计算机视觉, 载波芯片缺陷检测, 目标检测, YOLO-Efficientnet, 光发射次模块

Abstract: Chip on carrier（COC） is an important component of transmitter optical subassembly（TOSA） and is widely used in the field of optical communication to realize photoelectric conversion. Aiming at the problem of real-time detection of three different types of defects on COC, such as collapse, positioning column damage and waveguide stain, a defect detection algorithm of COC based on lightweight convolution neural network, YOLO-Efficientnet, is proposed. Firstly, in order to reduce network parameters and shorten detection time, lightweight convolution neural network, efficientnet, is used for image feature extraction as a backbone network. On the basis of MBConv, the attention idea of SENet is introduced, and the attention mechanism is introduced in the channel dimension. Secondly, in order to solve the problem of information loss during the downsampling process, the spatial pyramid pooling（SPP） structure is introduced to increase the receptive field of the image and separate more significant context features. Finally, to solve the problems of multi-scale COC defects and difficult detection of small targets in waveguide region, PANet structure is introduced for multi-scale feature fusion. Results of the experiments show that the algorithm proposed in this paper has an accuracy of 98.5% for COC defect detection, and the detection time reaches 0.42 seconds per picture, meeting the requirements of real-time detection.

Key words: computer vision, defect detection of chip on carrier, object detection, YOLO-Efficientnet, transmitter optical subassembly

周天宇, 朱启兵, 黄敏, 徐晓祥. 基于轻量级卷积神经网络的载波芯片缺陷检测[J]. 计算机工程与应用, 2022, 58(7): 213-219.

ZHOU Tianyu, ZHU Qibing, HUANG Min, XU Xiaoxiang. Defect Detection of Chip on Carrier Based on Lightweight Convolutional Neural Network[J]. Computer Engineering and Applications, 2022, 58(7): 213-219.

参考文献

[1] 黄健，张钢.深度卷积神经网络的目标检测算法综述[J].计算机工程与应用，2020，56（17）：12-23.
HUANG J，ZHANG G.Survey of object detection algorithms for deep convolutional neural networks[J].Computer Engineering and Applications，2020，56（17）：12-23.
[2] 罗会兰，彭珊，陈鸿坤.目标检测难点问题最新研究进展综述[J].计算机工程与应用，2021，57（5）：36-46.
LUO H，PENG S，CHEN H K，et al.Review on latest research progress of challenging problems in object detection[J].Computer Engineering and Applications，2021，57（5）：36-46.
[3] 方路平，何杭江，周国民.目标检测算法研究综述[J].计算机工程与应用，2018，54（13）：11-18.
FANG L P，HE H J，ZHOU G M.Research overview of object detection methods[J].Computer Engineering and Applications，2018，54（13）：11-18.
[4] 李章维，胡安顺，王晓飞.基于视觉的目标检测方法综述[J].计算机工程与应用，2020，56（8）：1-9.
LI Z W，HU A S，WANG X F.Survey of vision based object detection methods[J].Computer Engineering and Applications，2020，56（8）：1-9.
[5] GIRSHICK R，DONAHUE J，DARRELL T，et al.Rich feature hierarchies for accurate object detection and semantic segmentation[C]//Proceedings of 2014 IEEE Conference on Computer Vision and Pattern Recognition，June 23-28，2014：580-587.
[6] GIRSHICK R.Fast R-CNN[C]//Proceedings of 2015 IEEE International Conference on Computer Vision（ICCV），December 7-13，2015：1440-1448.
[7] REN S Q，HE K M，GIRSHICK R，et al.Faster R-CNN：Towards real-time object detection with region proposal networks[J].IEEE Transactions on Pattern Analysis and Machine Intelligence，2017，39（6）：1137-1149.
[8] HE K M，GKIOXARI G，DOLLAR P，et al.Mask R-CNN[J].IEEE Transactions on Pattern Analysis and Machine Intelligence，2020，42（2）：386-397.
[9] CAI Z W，VASCONCELOS N.Cascade R-CNN：Delving into high quality object detection[C]//IEEE Conference on Computer Vision and Pattern Recognition（CVPR），June 18-22，2018：6154-6162.
[10] SUN P，ZHANG R F，JIANG Y，et al.Y Sparse R-CNN：End-to-end object detection with learnable proposals[J].arXiv：2011.12450，2020.
[11] LIU W，ANGUELOV D，ERHAN D，et al.SSD：Single shot multibox detector[C]//Proceedings of European Conference on Computer Vision，2016：21-37.
[12] REDMON J，DIVVALA S，GIRSHICK R，et al.You only look once：Unified，real-time object detection[C]//Proceedings of 2016 IEEE Conference on Computer Vision and Pattern Recognition（CVPR），June 27-30，2016：6517-6525.
[13] REDMON J，FARHADI A.YOLO9000：Better，faster，stronger[C]//Proceedings of 2017 IEEE Conference on Computer Vision and Pattern Recognition（CVPR），July 21-26，2017：6517-6525.
[14] REDMON J，FARHADI A.YOLOV3：An incremental Improvement[J].arXiv：1804.02767，2018.
[15] BOCHKOVSKIY A，WANG C Y，LIAO H Y M.YOLOv4：Optimal speed and accuracy of object detection[J].arXiv：2004.10934，2020.
[16] 黄凤荣，李杨，郭兰申，等.基于Faster R-CNN的零件表面缺陷检测算法[J].计算机辅助设计与图形学学报，2020，32（6）：883-893.
HUANG F R，LI Y，GUO S L，et al.Method for detecting surface defects of engine parts based on faster R-CNN[J].Journal of Computer-Aided Design & Computer Graphics，2020，32（6）：883-893.
[17] CHEN S H，KANG C H，PERNG C B.Detecting and measuring defects in wafer die using GAN and YOLOv3[J].Applied Sciences，2020，10：8725.
[18] PEREZ H，TAH J H M，MOSAVI A.Deep learning for detecting building defects using convolutional neural networks[J].Sensors，2019，19（16）：3556.
[19] 金颖，王学影，段林茂.基于LabVIEW和Mask R-CNN的柱塞式制动主缸内槽表面缺陷检测[J].现代制造工程，2020（5）：125-132.
JIN Y，WANG X Y，DUAN L M.Surface defect detection of inner groove in plunger brake master cylinder based on LabVIEW and Mask R-CNN[J].Modern Manufacturing Engineering，2020（5）：125-132.
[20] TAN M，LE Q V.EfficientNet：Rethinking model scaling for convolutional neural networks[J].arXiv：1905.11946，2019.
[21] HE K M，ZHANG X Y，REN S Q，et al.Spatial pyramid pooling in deep convolutional networks for visual recognition[J].IEEE Transactions on Pattern Analysis and Machine Intelligence，2015，37（9）：1904-1916.
[22] LIU S，QI L，QIN H F，et al.Path aggregation network for instance segmentation[J].arXiv：1803.01534，2018.
[23] HE K M，ZHANG X Y，REN S Q，et al.Deep residual learning for image recognition[J].arXiv：1512.03385，2015.