A Comparative Analysis of Convolutional Neural Networks for Trash Classification

Authors

  • Md Atikuzzaman Department of Computer Science and Engineering, Green University of Bangladesh, Dhaka, Bangladesh
  • Md Parvez Hossain Department of Computer Science and Engineering, Green University of Bangladesh, Dhaka, Bangladesh
  • Md Zahidul Islam Department of Information and Communication Technology, Islamic University, Kustia, Bangladesh
  • Syed Ahsanul Kabir Department of Computer Science and Engineering, Green University of Bangladesh, Dhaka, Bangladesh

DOI:

https://doi.org/10.3329/gubjse.v8i1.62327

Keywords:

CNN, Trash Detection, ResNet, VGG, Trash Classi cation

Abstract

In the era of the twenty-first century, automation is the biggest fi eld of research. Although several works have been done successfully, there is still so much to do. Increasing trash could be the biggest threat to a better human life in the future. To automate the task, machines need to understand the type of trash to be recycled or separate similar trash for recycling. For this process, the perfect classi cation of trash plays a vital role in making a better life and a cleaner world. There are so many popular convolutional neural network (CNN) models for image classi cation. In this work, we examine and analyze the outcomes of several Residual Network(ResNet) and Visual Geometry Group (VGG) CNN models on a trash dataset. Our main investigation is the accuracy evaluation for different VGG and ResNet models where the training dataset, test dataset, number of epochs, and batch size are the same for all the models. Finally, we compare VGG and ResNet models with each other. We have got the peak accuracy for ResNet152 among all ResNet models and the peak accuracy on VGG16 among all VGG models. And we have got the maximum accuracy on ResNet152 among all the ResNet and VGG models which is about 94%.

GUB JOURNAL OF SCIENCE AND ENGINEERING, Vol 8, Dec 2021 P 17-23

Downloads

Download data is not yet available.
Abstract
108
PDF
127

Downloads

Published

2022-11-16

How to Cite

Atikuzzaman, M., Hossain, M. P. ., Islam, M. Z., & Kabir, S. A. (2022). A Comparative Analysis of Convolutional Neural Networks for Trash Classification. GUB Journal of Science and Engineering, 8(1), 17–23. https://doi.org/10.3329/gubjse.v8i1.62327

Issue

Section

Articles