Optics and Precision Engineering, Volume. 31, Issue 12, 1804(2023)

Multi-scale YOLOv5 for solar cell defect detection

Yafang CHEN1, Fei LIAO1、*, Xinyu HUANY1, Jing YANG2, and Hengxiang GONG1
Author Affiliations
  • 1College of Science, Chongqing University of Technology, Chongqing400054, China
  • 2Sichuan YC Garden Technology Co., Ltd, Yibin644000, China
  • show less
    Figures & Tables(17)
    Overall network architecture for the improved YOLOv5 algorithm
    Schematic of 3×3 DCNv2 layer
    Input and output feature maps for DCNv2 module
    Diagram of CA Module
    Input and output feature maps for CA module
    Structure of CA-PANet
    Fusion feature map of CA-PANet
    Raw image as well as vertically flipped, horizontally flipped, contrast enhanced and brightness enhanced images
    Raw EL image
    Comparison of AP and mAP between YOLOv5 before improvement and our YOLOv5
    Defect image detection with the addition of different improvement modules
    Defect detection results for different models of the same dataset
    Improved YOLOv5 model for common defect detection results
    • Table 1. Distribution table of the original data set

      View table
      View in Article

      Table 1. Distribution table of the original data set

      DatasetDefective imagesTotal
      CkBkCk&Bk
      Train2727857407
      Valid37121059
      Test782415117
      Total38711482583
    • Table 2. Distribution table of the amplified data set

      View table
      View in Article

      Table 2. Distribution table of the amplified data set

      Dataset-amplifiedDefective imagesTotal
      CkBkCk&Bk
      Train1 3603902852 035
      Valid37121059
      Test782415117
      Total1 4754263102 211
    • Table 3. Improved YOLOv5 model for ablation experiments

      View table
      View in Article

      Table 3. Improved YOLOv5 model for ablation experiments

      BaselineImproved StrategyGFLOPsParameters/MFPSmAP@0.5
      DCNv2CACross-level Concat
      YOLOv5-s---15.87.021110.924
      16.07.11660.933
      15.97.06780.936
      16.07.081090.915
      16.17.13750.939
      Ours16.37.22510.954
    • Table 4. Improved YOLOv5 model for ablation experiments

      View table
      View in Article

      Table 4. Improved YOLOv5 model for ablation experiments

      ModelGFLOPsParameters/MWeights/MFPSmAP@0.5
      YOLOv5-s15.87.0214.461110.924
      YOLOX-s26.68.9471.85530.940
      Ours16.37.2214.89510.954
    Tools

    Get Citation

    Copy Citation Text

    Yafang CHEN, Fei LIAO, Xinyu HUANY, Jing YANG, Hengxiang GONG. Multi-scale YOLOv5 for solar cell defect detection[J]. Optics and Precision Engineering, 2023, 31(12): 1804

    Download Citation

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

    Category: Information Sciences

    Received: Sep. 9, 2022

    Accepted: --

    Published Online: Jul. 25, 2023

    The Author Email: Fei LIAO (liaofei@cqut.edu.cn)

    DOI:10.37188/OPE.20233112.1804

    Topics