• 中国计算机学会会刊
  • 中国科技核心期刊
  • 中文核心期刊

Computer Engineering & Science

Previous Articles     Next Articles

Defect detection of mobile phone
motherboard based on RetinaNet

MA Mei-rong,LI Dong-xi     

  1. (School of Big Data,Taiyuan University of Technology,Jinzhong 030606,China)
  • Received:2019-10-12 Revised:2019-12-11 Online:2020-04-25 Published:2020-04-25

Abstract:

Since the motherboard images of different mobile phones have a multi-resolution imaging mode, the shape of defective components become multi-scale. Conventional defect detection methods mainly include image fusion methods or statistical model extraction methods, but the robustness of these methods still needs to be improved. To solve this problem, an automatic learning representation model, called RetinaNet object detector, is proposed. Firstly, feature pyramid network (FPN) is used to extract the multi-scale feature classification and location of defective components, and MobileNetV2 is introduced to compress and accelerate the RetinaNet model. Secondly, focus loss is used to resolve the class imbalance and increase the difficulty of detecting the contribution of the samples to the loss during the training. The experimental results show that RetinaNet can effectively detect defective components of different scales, and has high detection accuracy. Compared with other object detectors, RetinaNet achieves an average accuracy (mAP) of over 95%. These results demonstrate the effectiveness of the proposed model.
 

Key words: mobile phone motherboard, defect detection, RetinaNet, feature pyramid network (FPN), MobileNetV2, focus los