Steel Surface Defect Classification Via Deep Learning

Abstract

Deep learning and image processing methods have taken place in many parts of our lives, as well as in the quality control stages of production lines. The aim of this study is to train and use a deep learning model to improve quality management using limited data and computing power. To achieve that, deep learning for quality control models were trained by classifying six different steel surface defect images in the NEU-DET dataset. Xception, ResNetV2 152, VGG19 and InceptionV3 architectures were used to train the model. High accuracy was obtained with both Xception and ResNetV2 152.