Steel Surface Defect Detection Based on FF R-CNN

Abstract

A Faster R-CNN steel surface defect detection algorithm based on feature fusion and cascade detection network was proposed to solve the problem of low detection accuracy caused by reduced structure information when Deep Learning algorithm was used to detect steel surface defects. The improved Faster R-CNN algorithm is used to detect surface defects of steel. First, the feature map is extracted from the main network and fused to reduce the loss of structural information; Then the resulting feature map is further input into the RPN network generation area recommendation box; Finally, the detection network is used to classify and regress the regional recommendation box, and two detection networks are cascaded to achieve the target of accurate detection results. The model was analyzed by comparative experiments to find the algorithm model with the best detection accuracy. The proposed algorithm was tested on the NEU-DET dataset. The detection mean average precision of the backbone network using VGG-16 is 2.40% higher than that using Resnet-50. By fusing the features, the detection mean average precisionis improved by 11.86%. By detecting the cascade of the network, the detection mean average precision is improved by 2.37%. By continuously improving and optimizing the algorithm model, the detection mean average precision reaches 98.29%. Compared with traditional steel surface detection methods, this algorithm can detect the types and locations of steel surface defects more accurately, and improve the detection accuracy of steel surface defects.

Keywords

• surface defect detection of steel
• deep learning
• faster r-cnn
• feature fusion
• cascade detection network