Open Access iconOpen Access

ARTICLE

A Deep Trash Classification Model on Raspberry Pi 4

Thien Khai Tran1, Kha Tu Huynh2,*, Dac-Nhuong Le3, Muhammad Arif4, Hoa Minh Dinh1

1 Ho Chi Minh City University of Foreign Languages and Information Technology, Ho Chi Minh City, 700000, Vietnam
2 International University, Ho Chi Minh City, Vietnam-Vietnam National University, Ho Chi Minh City, 70000, Vietnam
3 Faculty of Information Technology, Haiphong University, Haiphong, 180000, Vietnam
4 Department of Computer Science and Information Technology, University of Lahore, Lahore, Pakistan

* Corresponding Author: Kha Tu Huynh. Email: email

Intelligent Automation & Soft Computing 2023, 35(2), 2479-2491. https://doi.org/10.32604/iasc.2023.029078

Abstract

Environmental pollution has had substantial impacts on human life, and trash is one of the main sources of such pollution in most countries. Trash classification from a collection of trash images can limit the overloading of garbage disposal systems and efficiently promote recycling activities; thus, development of such a classification system is topical and urgent. This paper proposed an effective trash classification system that relies on a classification module embedded in a hard-ware setup to classify trash in real time. An image dataset is first augmented to enhance the images before classifying them as either inorganic or organic trash. The deep learning–based ResNet-50 model, an improved version of the ResNet model, is used to classify trash from the dataset of trash images. The experimental results, which are tested both on the dataset and in real time, show that ResNet-50 had an average accuracy of 96%, higher than that of related models. Moreover, integrating the classification module into a Raspberry Pi computer, which controlled the trash bin slide so that garbage fell into the appropriate bin for inorganic or organic waste, created a complete trash classification system. This proves the efficiency and high applicability of the proposed system.

Keywords


Cite This Article

T. K. Tran, K. T. Huynh, D. Le, M. Arif and H. M. Dinh, "A deep trash classification model on raspberry pi 4," Intelligent Automation & Soft Computing, vol. 35, no.2, pp. 2479–2491, 2023.



cc This work is licensed under a Creative Commons Attribution 4.0 International License , which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.
  • 1623

    View

  • 515

    Download

  • 0

    Like

Share Link