A Steel Surface Defect Recognition Algorithm Based on Improved Deep Learning Network Model Using Feature Visualization and Quality Evaluation
Shengqi Guan, Ming Lei, Hao Lü
Abstract
Steel defect detection is used to detect defects on the surface of the steel and to improve the quality of the steel surface. However, traditional image detection algorithms cannot meet the detection requirements because of small defect features and low contrast between background and features about steel surface defect datasets. A novel recognition algorithm for steel surface defects based on improved deep learning network models using feature visualization and quality evaluation is proposed in this paper. Firstly, the VGG19 is used to pre-train the steel surface defect classification task and the corresponding DVGG19 is established to extract the feature images in different layers from defects weight model. Secondly, the SSIM and decision tree are used to evaluate the feature image quality and adjust the parameters and structure of VGG19. On this basis, a new VSD network is obtained and used for the classification of steel surface defects. Comparing with ResNet and VGG19 methods, experiment results show that the proposed method markedly can improve the average accuracy of classification, and the model is able to converge quickly, which can be good for steel surface defect recognition using VSD network model of feature visualization and quality evaluation.