Home > Archive > 2020 > Volume 10 Number 4 (July 2020) >
IJMLC 2020 Vol.10(4): 599-604 ISSN: 2010-3700
DOI: 10.18178/ijmlc.2020.10.4.979

A Web-Based Dataset for Garbage Classification Based on Shanghai’s Rule

Yunyi Liao

Abstract—In 2019, Shanghai has published a new regulation of trash management, which obtains a series of achievements in managing trash. In order to implement this regulation further and help citizens understand clearer about trash classification, we decided to develop a deep learning dataset and models to classify waste automatically. The object of this study is to take an image as input and identify the category of trash. We write a web crawler to capture images. After pre-processing, we gain about 14,000 images in total. We compare models including CNN, a ResNet50 model, and a VGG16 model on the dataset. Our experiments show that the ResNet50 model perform better than the others.

Index Terms—Dataset, garbage classification, web crawler, convolutional neural network, transfer learning.

The author is with the School of Information Science and Technology, Xiamen University Malaysia, Jalan Sunsuria, Bandar Sunsuria, 43900 Sepang, Selangor Darul Ehsan, Malaysia (e-mail: swe1709227@xmu.edu.my).


Cite: Yunyi Liao, "A Web-Based Dataset for Garbage Classification Based on Shanghai’s Rule," International Journal of Machine Learning and Computing vol. 10, no. 4, pp. 599-604, 2020.

Copyright © 2020 by the authors. This is an open access article distributed under the Creative Commons Attribution License which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited (CC BY 4.0).

General Information

  • ISSN: 2010-3700 (Online)
  • Abbreviated Title: Int. J. Mach. Learn. Comput.
  • Frequency: Bimonthly
  • DOI: 10.18178/IJMLC
  • Editor-in-Chief: Dr. Lin Huang
  • Executive Editor:  Ms. Cherry L. Chen
  • Abstracing/Indexing: Inspec (IET), Google Scholar, Crossref, ProQuest, Electronic Journals LibraryCNKI.
  • E-mail: ijmlc@ejournal.net

Article Metrics in Dimensions