Laser & Optoelectronics Progress, Volume. 59, Issue 18, 1815008(2022)

Detection Algorithm of Wire Harness Terminal Core Based on Improved EfficientDet

Shisong Zhu1、*, Xiushuai Sun1, Lishan Zhao1, Bibo Lu1, and Donglin Yao2
Author Affiliations
  • 1School of Computer Science and Technology, Henan Polytechnic University, Jiaozuo 454003, Henan , China
  • 2The First Military Representative Office of the Air Force Equipment Department in Wuhan, Wuhan 430000, Hubei , China
  • show less
    Figures & Tables(10)
    Core detection process of harness terminal microscopic image based on improved EfficientDet model
    Examples of anchor box generation. (a) 8×8 feature map; (b) 4×4 feature map
    Examples of data augmentation for microscopic images of wire harness terminals. (a) Original photo; (b) flip horizontal; (c) flip vertical; (d) saturation enhancement; (e) hue enhancement; (f) intensity enhancement; (g) salt and pepper noise; (h) Gaussian noise
    Learning rate and loss curves
    Comparison of detection results of different models
    P-R curves and area under the curve. (a) K-means; (b) GHM; (c) K-means and GHM
    Detection results of wire core of micrograph of wire harness terminals. (a) Harness terminal with 37 wire cores; (b) detection result of 37 cores terminal; (c) harness terminal with 84 wire cores; (d) detection result of 84 cores terminal
    Comparison of counting results between the proposed algorithm and manual method
    • Table 1. Number of images in the harness terminal microscopic image dataset

      View table

      Table 1. Number of images in the harness terminal microscopic image dataset

      DatasetOriginalData augmentationCore of cable
      Train385258277696
      Test96963074
      Total481267880770
    • Table 2. Performance comparison of single-stage target detection model on beam terminal micro image dataset

      View table

      Table 2. Performance comparison of single-stage target detection model on beam terminal micro image dataset

      Detection frameworkBackboneLoss typemAP /%Elapsed /sSpeed /(frame·s-1
      YOLOv3DarkNet-53CrossEntropy+MSE90.21.7355.61
      RetinaNetResNet-50-FPNFocalLoss+L1Loss89.82.7534.90
      NAS-FPNResNet-50-NAS-FPNFocalLoss+L1Loss89.54.1423.20
      EfficientDet-D0EfficientNet-B0-BiFPNFocalLoss+L1Loss90.71.5761.14
      Improved EfficientDet-D0EfficientNet-B0-BiFPNGHM-C+GHM-R96.21.6458.23
    Tools

    Get Citation

    Copy Citation Text

    Shisong Zhu, Xiushuai Sun, Lishan Zhao, Bibo Lu, Donglin Yao. Detection Algorithm of Wire Harness Terminal Core Based on Improved EfficientDet[J]. Laser & Optoelectronics Progress, 2022, 59(18): 1815008

    Download Citation

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

    Category: Machine Vision

    Received: Jul. 9, 2021

    Accepted: Aug. 2, 2021

    Published Online: Aug. 29, 2022

    The Author Email: Zhu Shisong (zss@hpu.edu.cn)

    DOI:10.3788/LOP202259.1815008

    Topics