TLU-Net: A Deep Learning Approach for Automatic Steel Surface Defect Detection

by   Praveen Damacharla, et al.

Visual steel surface defect detection is an essential step in steel sheet manufacturing. Several machine learning-based automated visual inspection (AVI) methods have been studied in recent years. However, most steel manufacturing industries still use manual visual inspection due to training time and inaccuracies involved with AVI methods. Automatic steel defect detection methods could be useful in less expensive and faster quality control and feedback. But preparing the annotated training data for segmentation and classification could be a costly process. In this work, we propose to use the Transfer Learning-based U-Net (TLU-Net) framework for steel surface defect detection. We use a U-Net architecture as the base and explore two kinds of encoders: ResNet and DenseNet. We compare these nets' performance using random initialization and the pre-trained networks trained using the ImageNet data set. The experiments are performed using Severstal data. The results demonstrate that the transfer learning performs 5 of the random initialization in defect classification. We found that the transfer learning performs 26 initialization in defect segmentation. We also found the gain of transfer learning increases as the training data decreases, and the convergence rate with transfer learning is better than that of the random initialization.


page 1

page 2

page 5


Using transfer learning to detect galaxy mergers

We investigate the use of deep convolutional neural networks (deep CNNs)...

Detection and Segmentation of Manufacturing Defects with Convolutional Neural Networks and Transfer Learning

Automatic detection of defects in metal castings is a challenging task, ...

Autonomous Underwater Robotic System for Aquaculture Applications

Aquaculture is a thriving food-producing sector producing over half of t...

Cost-based Feature Transfer for Vehicle Occupant Classification

Knowledge of human presence and interaction in a vehicle is of growing i...

TL-SDD: A Transfer Learning-Based Method for Surface Defect Detection with Few Samples

Surface defect detection plays an increasingly important role in manufac...

Deep Transfer-Learning for patient specific model re-calibration: Application to sEMG-Classification

Accurate decoding of surface electromyography (sEMG) is pivotal for musc...

An Automatic SOAP Classification System Using Weakly Supervision And Transfer Learning

In this paper, we introduce a comprehensive framework for developing a m...

Please sign up or login with your details

Forgot password? Click here to reset