Journal of Applied Optics, Volume. 45, Issue 4, 781(2024)

Defect detection of curved optical lenses based on improved YOLOv5s

Xiaolei LIU and Fenghui LIU*
Author Affiliations
  • School of Physics and Electronic Information, Henan Polytechnic University, Jiaozuo 454003, China
  • show less
    Figures & Tables(15)
    Images of lens defects
    Defect collection device
    Defect images generated by DCGAN
    Oversampling operation
    Structure diagram of YOLOv5s network
    SE module
    C3TR module
    Structure diagram of strengthen feature fusion
    Structure diagram of improved YOLOv5s model
    Comparison of mean average precision
    Comparison of defect detection results
    • Table 1. Experimental environment

      View table
      View in Article

      Table 1. Experimental environment

      操作系统显卡CUDACUDNN网络框架编程语言
      UbuntuRTX309011.08.0.4Pytorch1.7.0Python3.7
    • Table 2. YOLOv5s ablation experiment results

      View table
      View in Article

      Table 2. YOLOv5s ablation experiment results

      模型P/%R/%mAP/%AP/%
      xshy
      YOLOv5s91.288.290.791.188.092.890.9
      YOLOv5s+SE91.689.091.992.790.093.291.7
      YOLOv5s+SE+TR92.090.492.493.490.893.691.8
      YOLOv5s+SE+C291.789.792.994.290.594.192.8
      YOLOv5s+SE+TR+C292.291.693.994.492.195.094.1
    • Table 3. Comparison of improved network experiment results

      View table
      View in Article

      Table 3. Comparison of improved network experiment results

      模型YOLOv5sYOLOv8s
      P/%R/%mAP/%P/%R/%mAP/%
      原始91.288.290.791.185.490.0
      SE+TR+C292.291.693.992.986.591.5
    • Table 4. Comparative experimental results of different networks

      View table
      View in Article

      Table 4. Comparative experimental results of different networks

      模型t/msmAP/%AP/%
      xshy
      Mask R-CNN4177.576.781.267.584.7
      Faster R-CNN3774.960.486.478.674.2
      SSD2175.188.064.187.261.1
      改进YOLOv5s1493.994.492.195.094.1
      YOLOv7s1986.187.185.991.380.1
      YOLOx_s1789.589.388.290.589.8
    Tools

    Get Citation

    Copy Citation Text

    Xiaolei LIU, Fenghui LIU. Defect detection of curved optical lenses based on improved YOLOv5s[J]. Journal of Applied Optics, 2024, 45(4): 781

    Download Citation

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

    Category: Research Articles

    Received: Aug. 24, 2023

    Accepted: --

    Published Online: Oct. 21, 2024

    The Author Email: LIU Fenghui (刘丰慧)

    DOI:10.5768/JAO202445.0403003

    Topics