Implementation of Deep Learning Model for Classification of Household Trash Image

The problem of household waste management is a very important issue today, where the rapid urbanization, consumptive culture, and the tendency to dispose of waste without sorting it first from home, makes the volume of waste in landfills increase. Therefore, household waste management needs to be managed quickly and appropriately, so as not to have a major impact on environmental, hygiene, and health problems. Although some environmental communities and local governments have made efforts to manage waste through recycling systems, the long-term use of human labor is inefficient, expensive, and harmful to workers' health. Therefore, utilizing artificial intelligence technology is the best solution to classify waste types quickly and accurately. This research tries to test several pre-trained convolutional neural network (CNN) models to perform classification. The results of testing pre-trained CNN models, such as AlexNet, VGG16, VGG19, ResNet50, and ResNeXt50, found that the pre-trained model ResNext50 is better with 100% accuracy, while the training loss and validation loss are 0.0414 and 0.0304, respectively. Then the second best model is the pre-trained ResNet50 model with 100% accuracy with training loss and validation loss of 0.0832 and 0.1077, respectively.