Image Classification Comparison Using Neural Network and Support Vector Machine Algorithm With VGG16 As Feature Extraction Method

Authors

  • Aulia Wicaksono Universitas Udayana
  • I Putu Eka Nila Kencana Universitas Udayana
  • I Wayan Sumarjaya Universitas Udayana

DOI:

https://doi.org/10.62951/ijamc.v1i3.29

Keywords:

Image classification, Neural network, Support vector machine, VGG16

Abstract

Image classification is widely used in everyday life such as in car steering, closed-circuit television (CCTV), traffic cameras, etc. The implementation of image classification can be done using several methods, including neural network and support vector machine models. The neural network method is able to find the right weights that allow the network to show the desired behaviour while the support vector machine method has many dimensions and can overcome linear and non-linear data. In this research, feature extraction was carried out using VGG16 to increase accuracy. This research aims to find out how to implement the neural network and SVM algorithms to classify images and determine the results of analyzing the performance of the two methods. The data used in this study is secondary data consisting of 10 types of large wild cats with a total of 2339 training image datasets and 50 testing image datasets. The research stages consist of data augmentation, model design, model training, and model evaluation. Classification with the neural network model produced an accuracy of 96% and the support vector machine model produced an accuracy of 96%, which means that in a consistent training environment, the two models have the same performance.

References

Aggarwal, C. C. (2018). Neural networks and deep learning. Springer International Publishing. https://doi.org/10.1007/978-3-319-94463-0

Aremu, T., Zhiyuan, L., Alameeri, R., Khan, M., & El Saddik, A. (2023). SSIVD-Net: A novel salient super image classification & detection technique for weaponized violence. arXiv. https://arxiv.org/abs/2207.12850

Balestriero, R., Misra, I., & LeCun, Y. (2022). A data-augmentation is worth a thousand samples: Exact quantification from analytical augmented sample moments. arXiv. https://arxiv.org/abs/2202.08325

Buhot, A., & Gordon, M. B. (1998). Learning properties of support vector machines.

Chib, P. S., & Singh, P. (2023). Recent advancements in end-to-end autonomous driving using deep learning: A survey. arXiv. https://arxiv.org/abs/2307.04370

Dawson-Howe, K. (2014). A practical introduction to computer vision with OpenCV. Wiley.

Dede, D., Sarsıl, M. A., Shaker, A., Altıntaş, O., & Ergen, O. (2023). Next-gen traffic surveillance: AI-assisted mobile traffic violation detection system. arXiv. https://arxiv.org/abs/2311.16179

Goodfellow, I., Bengio, Y., & Courville, A. (2016). Deep learning. MIT Press.

Kumar, T., Mileo, A., Brennan, R., & Bendechache, M. (2023). Image data augmentation approaches: A comprehensive survey and future directions. arXiv. https://arxiv.org/abs/2301.02830

Murat, H. S. (2006). A brief review of feed-forward neural networks. Communications Faculty of Science University of Ankara, 50(1), 11–17. https://doi.org/10.1501/commua1-2_0000000026

O’Shea, K., & Nash, R. (2015). An introduction to convolutional neural networks.

Ojha, V. K., Abraham, A., & Snášel, V. (2017). Metaheuristic design of feedforward neural networks: A review of two decades of research. Engineering Applications of Artificial Intelligence, 60, 97–116. https://doi.org/10.1016/j.engappai.2017.01.013

Riera, I., Rana, A., & Lorente, Ò. (2021). Image classification with classic and deep learning techniques.

Simonyan, K., & Zisserman, A. (2014). Very deep convolutional networks for large-scale image recognition.

Sultana, F., Sufian, A., & Dutta, P. (2019). Advancements in image classification using convolutional neural network. Proceedings of the 2018 International Conference on Robotics, Control, and Information Technology (ICRCICN). https://doi.org/10.1109/ICRCICN.2018.8718718

Thai, L. H., Hai, T. S., & Thuy, N. T. (2012). Image classification using support vector machine and artificial neural network. International Journal of Information Technology and Computer Science, 4(5), 32–38. https://doi.org/10.5815/ijitcs.2012.05.05

Yang, S., Xiao, W., Zhang, M., Guo, S., Zhao, J., & Shen, F. (2022). Image data augmentation for deep learning: A survey.

Published

2024-10-14