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

    A surface defect detection method for curved optical lenses based on improved YOLOv5s was proposed to address the issue of low accuracy in surface defect detection. Firstly, a device for collecting surface defects on lenses was designed, and the collected defect images were enhanced to create a dataset of lens defects. Secondly, in order to enhance the utilization of channel information in YOLOv5s network, the SE attention mechanism was introduced into the feature extraction network to enable the network to extract information more accurately, and Transformer was integrated into the last C3 module of the backbone network to help the network better extract global information and improve detection efficiency. Finally, considering the issue of easy loss of feature information for small targets, the feature layer of the backbone network 160×160 was added to the feature fusion of the neck to increase the utilization of shallow information by the network. The mean average precision (mAP) and recall (R) of the improved YOLOv5s object detection algorithm are 93.9% and 91.6%, respectively, which are 3.2% and 3.4% higher than that of the original network algorithm, indicating that the improved YOLOv5s algorithm can effectively detect surface defects on lenses.

    Keywords
    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