The MobileOne-YOLO lightweight detection model is put forward for surface defect detection of wafer. Based on the YOLOv7 model, the backbone network is replaced with the MobileOne network. The re-parameterization and refocus convolution (RefConv) module and the coordinate attention (CA) mechanism are introduced into the feature pyramid network (FPN) to enhance feature extraction and strengthen inter-channel connections. As a result, the model performance is further improved. Ablation experiments on a self-collected material dataset validate that the improved algorithm surpasses current algorithms in both detection accuracy and speed.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Lightweight Vision-Based Wafer Defect Detection

  • Zilong Wang,
  • Lixin Lu,
  • Guiqin Li,
  • Peter Mitrouchev

摘要

The MobileOne-YOLO lightweight detection model is put forward for surface defect detection of wafer. Based on the YOLOv7 model, the backbone network is replaced with the MobileOne network. The re-parameterization and refocus convolution (RefConv) module and the coordinate attention (CA) mechanism are introduced into the feature pyramid network (FPN) to enhance feature extraction and strengthen inter-channel connections. As a result, the model performance is further improved. Ablation experiments on a self-collected material dataset validate that the improved algorithm surpasses current algorithms in both detection accuracy and speed.