Urban Solid Waste Classification Dataset for Convolutional Neural Networks. 2

  • Mirna Castro Bello Tecnológico Nacional de México
  • Víctor Manuel Romero Juárez Tecnológico Nacional de México
  • Jorge Fuentes Pacheco Tecnológico Nacional de México
  • Sergio Ricardo Zagal Barrera Tecnológico Nacional de México
  • Areli Bárcenas Nava Tecnológico Nacional de México
  • Diego Esteban Gutiérrez Valencia Tecnológico Nacional de México
Keywords: dataset, urban solid waste, convolutional neural networks, deep learning, classification

Abstract

Convolutional Neural Network (CNN) models require processed data that learns image patterns to avoid memorization. This research presents the creation of a dataset of 3,208 images for the classification of Urban Solid Waste (MSW) into organic and inorganic forms for CNN models. Processing was carried out using a three-phase methodology: 1. Dataset identification and selection: Kaggle and Github; 2. Dataset creation: image uniformity and color adjustments; and 3. Creation of the organic and inorganic waste dataset. The results obtained were the organic MSW datasets consisting of 1,574 images and 1,634 inorganic images. This will enable the training of Deep Learning models for binary MSW classification.

References

Adedeji, O., y Wang, Z. (2019). Intelligent Waste Classification System Using Deep Learning Convolutional Neural Network. Procedia Manufacturing, 35, 607–612. https://doi.org/10.1016/j.promfg.2019.05.086

Bricman, P. A., y Ionescu, R. T. (2018). CocoNet: A deep neural network for mapping pixel coordinates to color values. Lecture Notes in Computer Science (Including Subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics), 11302 LNCS, 64–76. https://doi.org/10.1007/978-3-030-04179-3_6

Ejaz, U., Hamza, M. A., y Kim, H. C. (2025). Channel Attention for Fire and Smoke Detection: Impact of Augmentation, Color Spaces, and Adversarial Attacks. Sensors 2025, Vol. 25, Page 1140, 25(4), 1140. https://doi.org/10.3390/S25041140

Kaza, S., Yao, L. C., Bhada-Tata, P., y Van Woerden, F. (2018). What a Waste 2.0: A Global Snapshot of Solid Waste Management to 2050. What a Waste 2.0: A Global Snapshot of Solid Waste Management to 2050. https://doi.org/10.1596/978-1-4648-1329-0

Kunwar, S. (2024). Garbage Dataset. Kaggle. https://doi.org/10.34740/KAGGLE/DSV/10182596

Liang, G., Panahi, F., Ahmed, A. N., Ehteram, M., Band, S. S., y Elshafie, A. (2021). Predicting municipal solid waste using a coupled artificial neural network with archimedes optimisation algorithm and socioeconomic components. Journal of Cleaner Production, 315, 128039. https://doi.org/10.1016/j.jclepro.2021.128039

Liu, X., Zhou, Q., Zhao, J., Shen, H., y Xiong, X. (2019). Fault Diagnosis of Rotating Machinery under Noisy Environment Conditions Based on a 1-D Convolutional Autoencoder and 1-D Convolutional Neural Network. Sensors, 19(4). https://doi.org/10.3390/s19040972

Mao, W. L., Chen, W. C., Wang, C. T., y Lin, Y. H. (2021). Recycling waste classification using optimized convolutional neural network. Resources, Conservation and Recycling, 164, 105132. https://doi.org/10.1016/J.RESCONREC.2020.105132

Mohamed, M. (2020). Garbage Classification (12 classes). https://www.kaggle.com/datasets/mostafaabla/garbage-classification

Namoun, A., Hussein, B. R., Tufail, A., Alrehaili, A., Syed, T. A., y BenRhouma, O. (2022). An Ensemble Learning Based Classification Approach for the Prediction of Household Solid Waste Generation. Sensors, 22(9). https://doi.org/10.3390/s22093506

Nnamoko, N., Barrowclough, J., y Procter, J. (2022). Solid Waste Image Classification Using Deep Convolutional Neural Network. Infrastructures, 7(4). https://doi.org/10.3390/infrastructures7040047

Ruiz, V., Sánchez, Á., Vélez, J. F., y Raducanu, B. (2019). Automatic Image-Based Waste Classification. In J. R. and de la P. L. F. and T. M. J. and A. H. Ferrández Vicente José Manuel and Álvarez-Sánchez (Ed.), From Bioinspired Systems and Biomedical Applications to Machine Learning (pp. 422–431). Springer International Publishing. https://doi.org/10.1007/978-3-030-19651-6_41

Sekar, S. (2018). Waste Classification data. https://www.kaggle.com/datasets/techsash/waste-classification-data

Single, S., Iranmanesh, S., y Raad, R. (2023). RealWaste: A Novel Real-Life Data Set for Landfill Waste Classification Using Deep Learning. Information, 14(12). https://doi.org/10.3390/info14120633

Srinilta, C., y Kanharattanachai, S. (2019). Municipal Solid Waste Segregation with CNN. 2019 5th International Conference on Engineering, Applied Sciences and Technology (ICEAST), 1–4. https://doi.org/10.1109/ICEAST.2019.8802522

Thung, G. (2017). garythung/trashnet: Dataset of images of trash; Torch-based CNN for garbage image classification. https://github.com/garythung/trashnet?tab=readme-ov-file

Xiaohoua. (2023). Garbage_HuaWei. https://www.kaggle.com/datasets/xiaohoua/garbage-huawei

Zhang, Q., Yang, Q., Zhang, X., Bao, Q., Su, J., y Liu, X. (2021). Waste image classification based on transfer learning and convolutional neural network. Waste Management, 135, 150–157. https://doi.org/10.1016/j.wasman.2021.08.038

Published
2025-05-01