A Mura detection model based on unsupervised adversarial learning

Abstract

Mura is a phenomenon in which the displays have various uneven display defects and has the characteristics of irregular shape and different sizes. For Mura detection, traditional detection methods have the following two problems: One is the problem of data imbalance. The second is that new shapes and sizes of Mura may appear at any time during the inspection process. In response to the above problems, this paper proposes the Res-unetGAN network, which is an unsupervised anomaly detection method based on generative adversarial network. The generative network is an autoencoder structure composed of ResNet50 and UNet to learn the normal distribution of normal samples. The discriminator is a convolutional neural network based on deep separable convolution and forms a game process with the generator. The network only needs normal samples during the training process, and the network is optimized by the error loss between the original samples and the reconstructed samples. In the test, a reconstruction error score will be designed according to the reconstruction quality, and the defect in the sample will be judged by the reconstruction error score, so as to achieve the goal of anomaly detection. After repeated experiments on the Mura data set, the detection accuracy of Mura defect is better than that of several models compared. The proposed model has a unique application prospect in other industrial anomaly detection since it only requires normal samples for training.