Defect detection algorithm based on deep neural network Mask R-CNN

Abstract

The invention discloses a defect detection algorithm based on a deep neural network Mask R-CNN, and belongs to the technical field of defect detection. The algorithm comprises the following specific steps of extracting features by using a feature pyramid network (FPN) based on ResNet50; extracting regions of interest (ROI) of a defect region by using a region proposal network (RPN) so as to obtaincorresponding anchor boxes; using a full convolutional neural network (FCN) to predict a pixel category in the ROI so as to realize defect segmentation; and finally, realizing prediction of the category to which each ROI belongs and corresponding anchor frame coordinates through a full connection layer of the network. Aiming at a magnetic shoe surface defect detection scene, the algorithm performs two improvements on a feature pyramid network (FPN) in MaskR-CNN: a C1 module is added in the FPN, and a pooling layer in a feature extraction layer of the C1 module is cancelled; and a CLAHE preprocessing module is added in front of a feature extraction layer of the FPN. Experimental results show that the algorithm of the invention has strong generalization ability and robustness, and can perform accurate defect segmentation on a magnetic shoe image.

Description

technical field [0001] The present invention relates to the technical field of defect detection, more specifically, to a defect detection algorithm based on deep neural network Mask R-CNN. Background technique [0002] With the proposal of "Made in China 2025" strategy and "Industry 4.0", China's traditional manufacturing industry is facing the huge challenge of industrial transformation and industrial upgrading, which promotes the development of products such as industrial robots, high-precision CNC machine tools and new energy vehicles And a wide range of applications, higher performance indicators are also proposed for the motor, and the surface quality of the magnetic tile directly affects the performance of the motor. Magnetic tile surface defect detection technology has the advantages of high detection efficiency, low cost, and high reliability. It is of great significance to the production of motors, and it also promotes the survival and development of enterprises. ...

AI Technical Summary

Problems solved by technology

[0004] The magnetic tile image has the characteristics of uneven illumination, complex surface texture, and low contrast. It is often difficult for traditional defect detection algorithms to accurately segment the defects.

Images