Author:
Abu M,Amir A,Lean Y H,Zahri N A H,Azemi S A
Abstract
Abstract
Detecting steel surface defects is one of the challenging problems for industries worldwide that had been used in manufacturing quality management. Manual inspection of the steel surface detects is a time-consuming process. This work aims at developing deep learning models that can perform steel defect detection and evaluating the potential of transfer learning for this task. In this paper, four types of transfer learning methods: ResNet, VGG, MobileNet, and DenseNet are experimentally evaluated to develop models for steel surface defect detection. The models were developed for binary classification (defect and no-defect) using the SEVERSTAL dataset from that contains 12,568 images of the steel surface. Then, these models were also assessed for multiclass classification using NEU dataset with 1800 images. In this work, image pre-processing is included to improve the result of steel defects detection. The experimental results have shown that the model developed by using the MobileNet method have the highest detection rate with 80.41% for the SEVERSTAL dataset and 96.94% for the NEU dataset compare to ResNet, VGG, and DenseNet transfer learning.
Subject
General Physics and Astronomy
Reference21 articles.
1. A Steel Surface Defect Recognition Algorithm Based on Improved Deep Learning Network Model Using Feature Visualization and Quality Evaluation;Guan;IEEE Access,2020
2. Detection of Steel Surface Defect Based on Machine Learning Using Deep Auto-encoder Network;Kholief,2017
3. Real-time Detection of Steel Strip Surface Defects Based on Improved YOLO Detection Network;Li;IFAC-PapersOnLine,2018
4. Striving for Simplicity: The All Convolutional Net;Springenberg,2015
Cited by
31 articles.
订阅此论文施引文献
订阅此论文施引文献,注册后可以免费订阅5篇论文的施引文献,订阅后可以查看论文全部施引文献