Evaluation of the convolutional neural network’s performance in classifying steel strip’s surface defects
Steel strip plays a vital role in many industrial fields. Its defects will impact the manifestation of the product and also reduce the features of the product, resulting in a huge economic loss. Deep learning algorithms, such as Convolutional Neural Network (CNN), have successfully been applied to i...
Saved in:
Main Authors: | , , , , , |
---|---|
Format: | Book Section |
Published: |
Springer Science and Business Media Deutschland GmbH
2022
|
Subjects: | |
Online Access: | http://eprints.utm.my/id/eprint/100487/ http://dx.doi.org/10.1007/978-981-33-4597-3_44 |
Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
id |
my.utm.100487 |
---|---|
record_format |
eprints |
spelling |
my.utm.1004872023-04-14T02:13:09Z http://eprints.utm.my/id/eprint/100487/ Evaluation of the convolutional neural network’s performance in classifying steel strip’s surface defects Tan, Kai Wen Mohd. Nor, Nur Safwati Fadil, Nor Akmal Mat Darus, Intan Zaurah Mohd. Yamin, Ahmad Hafizal Mohd. Zawawi, Fazila TJ Mechanical engineering and machinery Steel strip plays a vital role in many industrial fields. Its defects will impact the manifestation of the product and also reduce the features of the product, resulting in a huge economic loss. Deep learning algorithms, such as Convolutional Neural Network (CNN), have successfully been applied to image classification, while featuring a great level of abstraction and learning capabilities. These features are keys to detect and classify surface defects in a robust and reliable manner. The images used for training and testing the model are obtained from the NEU Surface Defect Database which contains six kinds of typical surface defects of steel strips that are rolled-in-scale, patches, crazing, pitted surface, inclusion and scratches. These images are pre-processing to enhance them and extract some useful information from them. After that, the CNN models are trained and tested with these images to evaluate their performance. The specific hyperparameters for the CNN model which are tuned are number of epochs, batch size, number of convolutional layers, input image size and kernel size. For each hyperparameter, the CNN model is trained and tested several times using different values of that hyperparameter. The training accuracy, testing accuracy and training time are recorded and analyzed. Lastly, the final CNN model with high performance is produced. The final CNN model based on the optimum hyperparameters is produced. It has a very high training accuracy of 95.12% and a fairly high testing accuracy of 85.43%. This paper focused on the application of CNN in the classification of the steel strip’s surface defects. The performance of the CNN models with different values of hyperparameters are also evaluated. Springer Science and Business Media Deutschland GmbH 2022 Book Section PeerReviewed Tan, Kai Wen and Mohd. Nor, Nur Safwati and Fadil, Nor Akmal and Mat Darus, Intan Zaurah and Mohd. Yamin, Ahmad Hafizal and Mohd. Zawawi, Fazila (2022) Evaluation of the convolutional neural network’s performance in classifying steel strip’s surface defects. In: Recent Trends in Mechatronics Towards Industry 4.0 Selected Articles from iM3F 2020, Malaysia. Lecture Notes in Electrical Engineering, 730 (NA). Springer Science and Business Media Deutschland GmbH, Singapore, pp. 485-495. ISBN 978-981334596-6 http://dx.doi.org/10.1007/978-981-33-4597-3_44 DOI:10.1007/978-981-33-4597-3_44 |
institution |
Universiti Teknologi Malaysia |
building |
UTM Library |
collection |
Institutional Repository |
continent |
Asia |
country |
Malaysia |
content_provider |
Universiti Teknologi Malaysia |
content_source |
UTM Institutional Repository |
url_provider |
http://eprints.utm.my/ |
topic |
TJ Mechanical engineering and machinery |
spellingShingle |
TJ Mechanical engineering and machinery Tan, Kai Wen Mohd. Nor, Nur Safwati Fadil, Nor Akmal Mat Darus, Intan Zaurah Mohd. Yamin, Ahmad Hafizal Mohd. Zawawi, Fazila Evaluation of the convolutional neural network’s performance in classifying steel strip’s surface defects |
description |
Steel strip plays a vital role in many industrial fields. Its defects will impact the manifestation of the product and also reduce the features of the product, resulting in a huge economic loss. Deep learning algorithms, such as Convolutional Neural Network (CNN), have successfully been applied to image classification, while featuring a great level of abstraction and learning capabilities. These features are keys to detect and classify surface defects in a robust and reliable manner. The images used for training and testing the model are obtained from the NEU Surface Defect Database which contains six kinds of typical surface defects of steel strips that are rolled-in-scale, patches, crazing, pitted surface, inclusion and scratches. These images are pre-processing to enhance them and extract some useful information from them. After that, the CNN models are trained and tested with these images to evaluate their performance. The specific hyperparameters for the CNN model which are tuned are number of epochs, batch size, number of convolutional layers, input image size and kernel size. For each hyperparameter, the CNN model is trained and tested several times using different values of that hyperparameter. The training accuracy, testing accuracy and training time are recorded and analyzed. Lastly, the final CNN model with high performance is produced. The final CNN model based on the optimum hyperparameters is produced. It has a very high training accuracy of 95.12% and a fairly high testing accuracy of 85.43%. This paper focused on the application of CNN in the classification of the steel strip’s surface defects. The performance of the CNN models with different values of hyperparameters are also evaluated. |
format |
Book Section |
author |
Tan, Kai Wen Mohd. Nor, Nur Safwati Fadil, Nor Akmal Mat Darus, Intan Zaurah Mohd. Yamin, Ahmad Hafizal Mohd. Zawawi, Fazila |
author_facet |
Tan, Kai Wen Mohd. Nor, Nur Safwati Fadil, Nor Akmal Mat Darus, Intan Zaurah Mohd. Yamin, Ahmad Hafizal Mohd. Zawawi, Fazila |
author_sort |
Tan, Kai Wen |
title |
Evaluation of the convolutional neural network’s performance in classifying steel strip’s surface defects |
title_short |
Evaluation of the convolutional neural network’s performance in classifying steel strip’s surface defects |
title_full |
Evaluation of the convolutional neural network’s performance in classifying steel strip’s surface defects |
title_fullStr |
Evaluation of the convolutional neural network’s performance in classifying steel strip’s surface defects |
title_full_unstemmed |
Evaluation of the convolutional neural network’s performance in classifying steel strip’s surface defects |
title_sort |
evaluation of the convolutional neural network’s performance in classifying steel strip’s surface defects |
publisher |
Springer Science and Business Media Deutschland GmbH |
publishDate |
2022 |
url |
http://eprints.utm.my/id/eprint/100487/ http://dx.doi.org/10.1007/978-981-33-4597-3_44 |
_version_ |
1764222573797703680 |
score |
13.18916 |