Comparison of CNN Models for Urban Garbage Image Classification in Thailand
Tanasit Tuangcharoentip, Nattee Niparnan · 2023
The authors believe that for society to continue improving, it needs to have a better waste classification system. This paper presents a computer vision model trained with a novel dataset comprising images of waste collected from Thailand, aimed at improving computer vision-based waste classification. We compare several CNN models, including VGG16, DenseNet169, ResNet-101, MobileNetV2, and InceptionV3 by using the Trashnet dataset developed by Gary Thung and our novel dataset. Our findings demonstrate that our model trained with both the Trashnet dataset and our new dataset outperforms the model trained by either dataset alone. The authors also discuss how this novel dataset can be improved to provide better results.