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

Visual detection for mobile phone surface defects based on semisupervised learning

  • Ge Yang,
  • Qifeng Zhou

摘要

This paper presents Fix-YOLOX (Fixmatch-You Only Look Once X), a semisupervised target detection model that uses a small amount of annotated data for fully supervised training, and adds a semisupervised training module using both pseudolabelling and consistent regularization to prevent overfitting in fully supervised training by using unlabelled data. Additionally, the generalization of the model and its fault tolerance to labelled data are improved. The experimental results show that the proposed semisupervised visual detection algorithm, Fix-YOLOX, can substantially reduce the amount of data annotation required for the target detection task while effectively overcoming the problem caused by annotated data with uneven quality. The YOLOX model achieves 91.95% accuracy with 50% annotated data and an average detection time of 10.4 ms per image/frame, which is consistent with the detection time of original YOLOX. Therefore, the model has good real-time performance and generalizability.