Abstract
Surface defect detection is very important to ensure product quality, but most of the surface defects of industrial products are characterized by low contrast, big size difference and category similarity, which brings challenges to the automatic detection of defects. To solve these problems, we propose a defect detection method based on convolutional neural network. In this method, a backbone network with semantic supervision is applied to extract the features of different levels. While a multi-level feature fusion module is proposed to fuse adjacent feature maps into high-resolution feature maps successively, which significantly improves the prediction accuracy of the network. Finally, an Encoding module is used to obtain the global context information of the high-resolution feature map, which further improves the pixel classification accuracy. Experiments show that the proposed method is superior to other methods in NEU_SEG (mIoU of 85.27) and MT (mIoU of 77.82) datasets, and has the potential of real-time detection.
PCB Electronic Component Defect Detection Method based on Improved YOLOv4 Algorithm
