Image Classification of Room Tidiness Using VGGNet with Data Augmentation

Leni Fitriani; Ayu Latifah; Moch. Rizky Cahyadiputra

doi:10.30595/juita.v12i1.21204

Authors

Leni Fitriani Department of Computer Science, Institut Teknologi Garut http://orcid.org/0000-0002-9035-5862
Ayu Latifah Department of Computer Science, Institut Teknologi Garut
Moch. Rizky Cahyadiputra Department of Computer Science, Institut Teknologi Garut

DOI:

https://doi.org/10.30595/juita.v12i1.21204

Keywords:

Convolutional Neural Network, data augmentation, image classification, room tidiness, VGGNet

Abstract

Tidiness becomes an essential aspect that everyone should maintain. Tidiness encompasses various elements, and one of the aspects closely related to it is the tidiness of a room. The tidiness of a room creates a comfortable and clean environment. The tidiness of a room is particularly crucial for individuals involved in businesses such as the hospitality industry. Therefore, a solution is needed to address this issue, and one of the approaches is to utilize Deep Learning for automatic room tidiness classification. One popular deep learning method for implementing image classification of room tidiness is the convolutional neural network (CNN), which creates a well-performing model for image classification with data augmentation. This research aims to develop an image classification model using CNN with the VGGNet architecture and data augmentation. This study is a reference for further development, with potential applications in the hospitality industry. The research results in a model that achieves an accuracy of 98.44% with a data proportion of 90% for training and validation, while the remaining 10% is used for testing purposes. The conclusion drawn from this study is that the CNN method, combined with data augmentation, can be utilized for image classification of room tidiness.

References

[1] D. Collins, “The Health Benefits of Neatness and Order,” Linkedin, Mar. 2020.

[3] K. Cherry, “What Does a Messy Room Say About You,” verywellmind, Feb. 2021.

[4] A. Latifah, S. H. Supangkat, and A. Ramelan, “Smart Building: A Literature Review,” in 2020 International Conference on ICT for Smart Society (ICISS), Nov. 2020, pp. 1–6, doi: 10.1109/ICISS50791.2020.9307552.

[5] M. J. Dharmali, T. Lioner, and V. V. Susilo, “SISTEM KLASIFIKASI KERAPIHAN KAMAR HOTEL MENGGUNAKAN CONVOLUTED NEURAL NETWORK (CNN),” Comput. J. Comput. Sci. Inf. Syst., vol. 5, no. 2, p. 61, Dec. 2021, doi: 10.24912/computatio.v5i2.15175.

[6] L. Alzubaidi, “Review of deep learning: concepts, CNN architectures, challenges, applications, future directions,” J. Big Data, vol. 8, no. 1, p. 53, Mar. 2021, doi: 10.1186/s40537-021-00444-8.

[7] S. Raschka and V. Mirjalili, Machine Learning and Deep Learning with Python, scikit-learn, and TensorFlow 2, 3rd ed. Birmingham B3 2PB, UK: Packt Publishing Ltd., 2019.

[8] K. Singh, D. Singh, and N. Mishra, “Review: Convolutional neural networks and its architecture,” Int. J. Health Sci. (Qassim)., May 2022, doi: 10.53730/ijhs.v6nS1.7074.

[9] U. Muhammad, W. Wang, S. P. Chattha, and S. Ali, “Pre-trained VGGNet Architecture for Remote-Sensing Image Scene Classification,” in 2018 24th International Conference on Pattern Recognition (ICPR), Aug. 2018, pp. 1622–1627, doi: 10.1109/ICPR.2018.8545591.

[10] C. Shorten and T. M. Khoshgoftaar, “A survey on Image Data Augmentation for Deep Learning,” J. Big Data, vol. 6, no. 1, p. 60, Dec. 2019, doi: 10.1186/s40537-019-0197-0.

[11] S. Suparni, H. Rachmi, and A. Al Kaafi, “Detection of Room Cleanliness Based on Digital Image Processing using SVM and NN Algorithm,” SinkrOn, vol. 7, no. 3, pp. 777–783, Jul. 2022, doi: 10.33395/sinkron.v7i3.11479.

[12] A. Mahajan and S. Chaudhary, “Categorical Image Classification Based On Representational Deep Network (RESNET),” in 2019 3rd International conference on Electronics, Communication and Aerospace Technology (ICECA), Jun. 2019, pp. 327–330, doi: 10.1109/ICECA.2019.8822133.

[13] S. Wan, Y. Liang, and Y. Zhang, “Deep convolutional neural networks for diabetic retinopathy detection by image classification,” Comput. Electr. Eng., vol. 72, pp. 274–282, Nov. 2018, doi: 10.1016/j.compeleceng.2018.07.042.

[14] L. Fitriani, D. Tresnawati, and M. B. Sukriyansah, “Image Classification On Garutan Batik Using Convolutional Neural Network with Data Augmentation,” JUITA J. Inform., vol. 11, no. 1, p. 107, 2023, doi: 10.30595/juita.v11i1.16166.

[15] S. S. Yadav and S. M. Jadhav, “Deep convolutional neural network based medical image classification for disease diagnosis,” J. Big Data, vol. 6, no. 1, p. 113, Dec. 2019, doi: 10.1186/s40537-019-0276-2.