Mobile phone glass defect detection method and device, computer equipment and storage medium

Abstract

The invention relates to a mobile phone glass defect detection method and device, computer equipment and a storage medium. The mobile phone glass defect detection method comprises the following steps: acquiring original defect image data of mobile phone cover plate glass; expanding the original defect image data by using a WGAN-GP network to obtain expanded defect image data; obtaining a VOC defect data set of the mobile phone cover plate glass; building an improved YOLOv3 detection network; clustering the mark box by using a K-means algorithm to obtain a plurality of anchor boxes, and adding the anchor boxes into a detection layer; performing basic training on the VOC defect data set by using an improved YOLOv3 detection network to obtain a weight file; performing sparse training on the weight file according to the weight coefficient of the BN layer in the improved YOLOv3 detection network to obtain a weight and a network framework; and testing the detection data set for the weight and the network framework to obtain category information, position information and confidence information of the detection target. The method has high defect detection efficiency and high defect detection precision at the same time.

Description

technical field [0001] The invention relates to the field of communication technology, in particular to a method, device, computer equipment and storage medium for detecting defects of mobile phone glass. Background technique [0002] With the advent of the 5G era, smartphones have increasingly become an indispensable communication tool for people. As the communication medium between the mobile phone and the user, the glass screen of the mobile phone affects the quality of the user's experience in its appearance. During the processing of mobile phone cover glass, defects such as dots and lines will occur, which seriously affect the quality of mobile phone glass screens. [0003] The traditional methods for detecting glass defects include manual inspection methods, glass defect detection methods based on deep learning methods, defect detection methods based on traditional template matching methods, and glass defect detection methods based on deep learning target detection me...

AI Technical Summary

Problems solved by technology

[0006] Based on this, it is necessary to provide a mobile phone glass defect detection method, device, computer equipment and storage medium that can improve the detection efficiency and detection accuracy of glass defects in view of the problems of low detection efficiency and detection accuracy in traditional glass detection methods

Images