Improving Resnet Model In Safety Gear Classification Using Finest Optimizer

Robet; Johanes Terang Kita Perangin Angin; Edi Wijaya

doi:10.59934/jaiea.v4i2.703

Authors

Robet STMIK TIME
Johanes Terang Kita Perangin Angin STMIK TIME
Edi Wijaya STMIK TIME

DOI:

https://doi.org/10.59934/jaiea.v4i2.703

Keywords:

safety gear; classification; resnet; pre-trained; optimizer

Abstract

The Occupational accidents that occur in the work environment are increasing day by day. This is caused by workers' non-compliance with the established work safety equipment. Although the supervision of the use of work safety equipment has been carried out, it is still done manually involving less effective human resources. Therefore, it is necessary to develop an intelligent model that can classify the use of work safety equipment more accurately. This study uses the pre-trained ResNet50 model and is combined with the best optimization model to improve accuracy. The results of the study showed that the RMSProp optimization model has better performance with an accuracy value of 97.01% in the 17th epoch of 50 epochs of data training and with training loss and validation loss values of 0.3268 and 0.145, respectively. Testing of 20 images with each image, 10 images using safety equipment, and 10 images not using safety equipment can be classified correctly.

Downloads

Download data is not yet available.

References

B. Ketenagakerjaan, “Kecelakaan Kerja Makin Marak Lima Tahun Terakhir,” Bpjsketenagakerjaan.Go.Id, 2024. https://www.bpjsketenagakerjaan.go.id/berita/28681/Kecelakaan-Kerja-Makin-Marak-dalam-Lima-Tahun-Terakhir

K. Nisa, F. Nur Fajri, and Z. Arifin, “Implementation of Personal Protective Equipment Detection Using Django and Yolo Web at Paiton Steam Power Plant (PLTU),” J. Ilm. Tek. Elektro Komput. dan Inform., vol. 9, no. 2, pp. 333–347, 2023, doi: 10.26555/jiteki.v9i2.26131.

H. Wang, “Detection of Personal Protective Equipment (PPE) using an Anchor Free-Convolutional Neural Network,” Int. J. Adv. Comput. Sci. Appl., vol. 15, no. 2, pp. 366–374, 2024, doi: 10.14569/IJACSA.2024.0150239.

M. I. B. Ahmed et al., “Personal Protective Equipment Detection: A Deep-Learning-Based Sustainable Approach,” Sustain., vol. 15, no. 18, pp. 1–18, 2023, doi: 10.3390/su151813990.

A. Barlybayev et al., “Personal protective equipment detection using YOLOv8 architecture on object detection benchmark datasets: a comparative study,” Cogent Eng., vol. 11, no. 1, p., 2024, doi: 10.1080/23311916.2024.2333209.

K. O. P. P. Nugraha and A. P. Rifai, “Convolutional Neural Network for Identification of Personal Protective Equipment Usage Compliance in Manufacturing Laboratory,” J. Ilm. Tek. Ind., vol. 22, no. 1, pp. 11–24, 2023, doi: 10.23917/jiti.v22i1.21826.

A. Krizhevsky, I. Sutskever, and G. E. Hinton, “ImageNet classification with deep convolutional neural networks,” Adv. Neural Inf. Process. Syst., vol. 2, pp. 1097–1105, 2012.

Y. Li, H. Wei, Z. Han, J. Huang, and W. Wang, “Deep Learning-Based Safety Helmet Detection in Engineering Management Based on Convolutional Neural Networks,” Adv. Civ. Eng., vol. 2020, 2020, doi: 10.1155/2020/9703560.

Robet, J. Terang Kita Perangin Angin, and O. Pribadi, “Implementation of Deep Learning Model for Classification of Household Trash Image,” vol. 8, no. October, pp. 2575–2583, 2024.

R. S. Pothineni, S. Inampudi, L. Y. Gudavalli, and T. Lakshmi Surekha, “Traffic Sign Classification using Deep Learning,” Proc. 3rd Int. Conf. Artif. Intell. Smart Energy, ICAIS 2023, pp. 527–531, 2023, doi: 10.1109/ICAIS56108.2023.10073690.

S. Sethi, M. Kathuria, and T. Kaushik, “Face mask detection using deep learning: An approach to reduce risk of Coronavirus spread,” J. Biomed. Inform., vol. 120, 2021, doi: 10.1016/j.jbi.2021.103848.

S. Shomal Zadeh, S. Aalipour birgani, M. Khorshidi, and F. Kooban, “Concrete Surface Crack Detection with Convolutional-based Deep Learning Models,” SSRN Electron. J., vol. 10, no. 3, pp. 25–35, 2024, doi: 10.2139/ssrn.4661249.

S. Z. N. A. Silopung, R. Samad, M. Mustafa, N. R. H. Abdullah, and N. Fadilah, “Analysis of Personal Protective Equipment Classification Method Using Deep Learning,” IET Conf. Proc., vol. 2022, no. 22, pp. 347–355, 2022, doi: 10.1049/icp.2022.2643.

K. He, X. Zhang, S. Ren, and J. Sun, “Deep residual learning for image recognition,” Proc. IEEE Comput. Soc. Conf. Comput. Vis. Pattern Recognit., vol. 2016-Decem, pp. 770–778, 2016, doi: 10.1109/CVPR.2016.90.

M. Nain, S. Sharma, and S. Chaurasia, “Safety and Compliance Management System Using Computer Vision and Deep Learning,” IOP Conf. Ser. Mater. Sci. Eng., vol. 1099, no. 1, p. 012013, 2021, doi: 10.1088/1757-899x/1099/1/012013.

K. A. Rahman, “Is an Employee Wearing Safety Gear?” https://www.kaggle.com/datasets/khananikrahman/is-an-employee-wearing-safety-gear