Laser & Optoelectronics Progress, Volume. 59, Issue 24, 2415006(2022)

Real-Time Optical Fiber End Surface Defects Detection Model Based on Lightweight Improved Network

Minyu Song1, Lirong Chen1、*, Jian'an Liang1, Jinpeng Li1, Zhenzhen Niu1, Zhen Wang1, and Lili Bai2
Author Affiliations
  • 1College of Physics and Electronic Engineering, Shanxi University, Taiyuan 030006, Shanxi, China
  • 2College of Aeronautics and Astronautics, Taiyuan University of Technology, Taiyuan 030006, Shanxi, China
  • show less

    Optical fiber is an indispensable transmission medium in modern communication system and quantum secure communication network. In order to solve the problem that the optical fiber end surface defects cause transmission quality decline or even permanent damage to optical transmission system, a fiber end surface detection model YOLOv5_CS based on YOLOv5 algorithm is proposed. Firstly, ShuffleNetV2, a lightweight network, is used as the main feature extraction network. Deep convolution operation and channel random mixing strategy are used to reduce model capacity and enrich feature information. Secondly, the convolutional block attention module (CBAM) is introduced, and features are enhanced in both spatial and channel dimensions to improve network performance. Finally, the number of convolution kernels in the feature fusion layer is reduced to achieve further model compression. The validity of the proposed method is compared and verified by the optical fiber end data set constructed by data augmentation technology. The results show that compared with the YOLOv5 algorithm, the model capacity of the proposed model is reduced by 80%, the detection speed is increased by 31.1 frame/s, and the mean average precision (mAP) is increased by 1.7%, which can accurately and real-time detect optical fiber end surface defects. This work is aimed at the development of portable intelligent detection device, and can also provide technical support for optical fiber end surface defects detection and related visual sensing industry.

    Tools

    Get Citation

    Copy Citation Text

    Minyu Song, Lirong Chen, Jian'an Liang, Jinpeng Li, Zhenzhen Niu, Zhen Wang, Lili Bai. Real-Time Optical Fiber End Surface Defects Detection Model Based on Lightweight Improved Network[J]. Laser & Optoelectronics Progress, 2022, 59(24): 2415006

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Machine Vision

    Received: Sep. 30, 2021

    Accepted: Nov. 1, 2021

    Published Online: Nov. 28, 2022

    The Author Email: Chen Lirong (clr@sxu.edu.cn)

    DOI:10.3788/LOP202259.2415006

    Topics