Small metal objects classification based on the deep learning approach
2024,
Nur Safariah Inani M. Tahir,
Masturah Tunnur Mohamad Talib,
Mohd Hafiz Fazalul Rahiman,
N. S. Khalid,
Siti Marhainis Othman
Classification of small metal objects plays a crucial role in various engineering fields, including manufacturing, robotics, and security. With advancements in deep learning techniques, the use of Convolutional Neural Networks (CNNs) has emerged as a powerful tool for image classification tasks. The methodology begins by collecting diverse datasets consisting of images of small metal objects. The datasets are labelled with corresponding object classes to facilitate supervised learning. Preprocessing techniques like re-sizing, normalization, and augmentation are used to improve the quality and diversity of the datasets. The use of CNNs in classification can be a better option compared to commonly used machine learning approaches. The CNN architecture is designed and trained to learn the distinguishing features of small metal objects. The main objective of this study is to assess the accuracy of this classification and explain how CNNs can enhance classification accuracy. The results of this study also show the effect of the optimizer on the classification process, which changes when different types of optimizers such as RMSprop, Adam, and SGD are used. While some optimizers yield slightly lower accuracy results, the Adam optimizer with the CNN ResNet-50 module proves suitable for use with this dataset, achieving a high classification accuracy of 86%.
