Skip to main content
Article | Green Intelligent Systems and Applications | Volume 5 - Issue 1 - 2025 | 93-105 | https://doi.org/10.53623/gisa.v5i1.581
© 2025 by the author(s), and is licensed under the Creative Commons Attribution 4.0 International (CC BY 4.0) License Creative Commons Attribution 4.0 International (CC BY 4.0) License

Classification of Metal Surface Defects Using Convolutional Neural Networks (CNN)

Author(s): Dhika Wahyu Pratama 1 , Muchammad Ismail 1 , Restu Nurraudah 1 , Achmad Pratama Rifai 1 ORCID https://orcid.org/0000-0003-4890-8344 , Huu Tho Nguyen 2 ORCID https://orcid.org/0000-0001-8558-5906
Author(s) information:
1 Department of Mechanical and Industrial Engineering, Universitas Gadjah Mada, Yogyakarta, Indonesia
2 Faculty of Engineering and Technology, Nguyen Tat Thanh University, Ho Chi Minh City, Vietnam

Corresponding author

Green Intelligent Systems and Applications

Volume 5 - Issue 1 - 2025

 About the journal
Submit your article

Metal surface quality inspection is an important step in ensuring that products meet predetermined industry standards. The manual methods used were often slow and prone to errors, so more efficient solutions were needed. The application of Machine Learning (ML)-based technologies, especially Convolutional Neural Networks (CNN), offered an innovative approach to overcome these challenges. CNN had the ability to automatically extract visual features from images with high accuracy, making it an effective tool in defect classification. This research used several CNN architectures, including MobileNetV2 and InceptionV3, as well as a model developed in-house, the K3 Model. Data augmentation, such as rotation and lighting adjustments, was applied to increase variation in the dataset and aid the model in generalization. The research results showed that the K3+Augmentation model achieved the highest accuracy of 100% in testing, with a very low loss of 0.0009. While MobileNetV2 offered better training speed, K3+Augmentation showed superior performance in detecting and classifying metal defects. These findings confirmed the potential of CNN in improving the efficiency of quality inspection in modern industry.

Read Full-Text
Previous article
Next article

Cuan-Urquizo, E.; Barocio, E.; Tejada-Ortigoza, V.; Pipes, R. B.; Rodríguez, C. A.; Roman-Flores, A. (2019). Characterization of the mechanical properties of FFF structures and materials: A review on experimental, computational, and theoretical approaches. Materials, 12, 895. https://doi.org/10.3390/ma12060895.

Zaman, U.K.; Boesch, E.; Siadat, A.; Rivette, M.; Baqai, A.A. (2018). Impact of fused deposition modeling (FDM) process parameters on strength of built parts using Taguchi’s design of experiments. The International Journal of Advanced Manufacturing Technology, 101, 1215–1226. https://doi.org/10.1007/s00170-018-3014-6.

Sandler, M.; Howard, A.; Zhu, M.; Zhmoginov, A.; Chen, L.C. (2018). MobileNetV2: Inverted Residuals and Linear Bottlenecks. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 4510–4520. https://doi.org/10.1109/CVPR.2018.00474.

Szegedy, C.; Vanhoucke, V.; Ioffe, S.; Shlens, J.; Wojna, Z. (2016). Rethinking the Inception Architecture for Computer Vision. Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2818–2826. https://doi.org/10.1109/CVPR.2016.308.

Li, J.; Su, Z.; Geng, J.; Yin, Y. (2018). Real-time detection of steel strip surface defects based on improved YOLO detection network. IFAC-PapersOnLine, 51, 76–81. https://doi.org/10.1016/j.ifacol.2018.09.412.

Shorten, C.; Khoshgoftaar, T.M. (2019). A survey on image data augmentation for deep learning. Journal of Big Data, 6, 60. https://doi.org/10.1186/s40537-019-0197-0.

Guan, S.; Lei, M.; Lu, H. (2020). A steel surface defect recognition algorithm based on improved deep learning network model using feature visualization and quality evaluation. IEEE Access, 8, 49885–49895. https://doi.org/10.1109/ACCESS.2020.2979755.

Babbie, E. R. (2020). The Practice of Social Research; Cengage Au: Boston, USA.

Goodfellow, I.; Bengio, Y.; Courville, A. (2016). Deep Learning; MIT Press: Cambridge, USA.

Russakovsky, O.; Deng, J.; Su, H.; Krause, J.; Satheesh, S.; Ma, S.; Huang, Z.; Karpathy, A.; Khosla, A.; Bernstein, M.; Berg, A.C.; Fei-Fei, L. (2015). ImageNet Large Scale Visual Recognition Challenge. International Journal of Computer Vision, 115, 211–252. https://doi.org/10.1007/s11263-015-0816-y.

Zhao, Z.Q.; Zheng, P.; Xu, S.T.; Wu, X. (2019). Object detection with deep learning: A review. IEEE Transactions on Neural Networks and Learning Systems, 30, 3212–3232. https://doi.org/10.1109/TNNLS.2018.2876865.

Han, J.; Pei, J.; Tong, H. (2022). Data Mining: Concepts and Techniques, 3rd ed.; Morgan Kaufmann: Burlington, USA.

Perez, L.; Wang, J. (2017). The effectiveness of data augmentation in image classification using deep learning. https://doi.org/10.48550/arXiv.1712.04621

Zhang, H.; Cisse, M.; Dauphin, Y. .; Lopez-Paz, D. (2017). Mixup: Beyond empirical risk minimization. https://doi.org/10.48550/arXiv.1710.09412.

Wong, S.C.; Gatt, A.; Stamatescu, V.; McDonnell, M.D. (2016). Understanding data augmentation for classification: When to warp? 2016 International Conference on Digital Image Computing: Techniques and Applications (DICTA), 1–6. https://doi.org/10.1109/DICTA.2016.7797091.

Rasyidi, M.A.; Bariyah, T. (2020). Batik pattern recognition using convolutional neural network. Bulletin of Electrical Engineering and Informatics, 9, 1430–1437. https://doi.org/10.11591/eei.v9i4.2385.

Tan, M.; Le, Q. (2019). EfficientNet: Rethinking model scaling for convolutional neural networks. International Conference on Machine Learning, 6105–6114.

Kingma, D.P.; Ba, J. (2014). Adam: A method for stochastic optimization. https://doi.org/10.48550/arXiv.1412.6980.

Islam, M.F.; Rahman, M.M. (2018). Metal surface defect inspection through deep neural network. 2018 International Conference on Mechanical, Industrial and Energy Engineering (ICMIEE), 258. https://api.semanticscholar.org/CorpusID:235365832.

Majeed, A.P.A.; Abdullah, M.A.; Nasir, A.F.A.; Razman, M.A.M.; Chen, W.; Yap, E.H. (2024). Surface defect detection: A feature-based transfer learning approach. Journal of Physics: Conference Series, 2762, 012088. https://doi.org/10.1088/1742-6596/2762/1/012088.

Read Full-Text
About this article

SUBMITTED: 07 January 2025
ACCEPTED: 14 April 2025
PUBLISHED: 3 May 2025
SUBMITTED to ACCEPTED: 98 days
DOI: https://doi.org/10.53623/gisa.v5i1.581

Cite this article
Pratama, D. W. ., Ismail, M. ., Nurraudah, R. ., Rifai, A. P., & Nguyen , H. T. (2025). Classification of Metal Surface Defects Using Convolutional Neural Networks (CNN). Green Intelligent Systems and Applications, 5(1), 93–105. https://doi.org/10.53623/gisa.v5i1.581
Keywords
Metal Surface Inspection CNN MobileNetV2 K3 Model InceptionV3 Augmentation
Accessed
107
Citations
0
Share this article