期刊
RESOURCES CONSERVATION AND RECYCLING
卷 164, 期 -, 页码 -出版社
ELSEVIER
DOI: 10.1016/j.resconrec.2020.105132
关键词
Convolutional neural network (CNN); Genetic algorithm (GA); DenseNet; Waste classification; Image recognition; Recycling
资金
- National Science Council of Taiwan [MOST 108-2218-E-224 -004 -MY3]
This study discussed the application of an automatic classification robot based on image recognition in recycling tasks, optimizing the fully-connected layer of DenseNet121 to improve waste classification accuracy, and utilizing genetic algorithms for optimization. Results showed that the optimized DenseNet121 achieved a highest accuracy of 99.6% on TrashNet, outperforming other studies' CNNs.
An automatic classification robot based on effective image recognition could help reduce huge labors of recycling tasks. Convolutional neural network (CNN) model, such as DenseNet121, improved the traditional image recognition technology and was the currently dominant approach to image recognition. A famous benchmark dataset, i.e., TrashNet, comprised of a total of 2527 images with six different waste categories was used to evaluate the CNNs' performance. To enhance the accuracy of waste classification driven by CNNs, the data augmentation method could be adopted to do so, but fine-tuning optimally hyper-parameters of CNN's fullyconnected-layer was never used. Therefore, besides data augmentation, this study aims to utilize a genetic algorithm (GA) to optimize the fully-connected-layer of DenseNet121 for improving the classification accuracy of DenseNet121 on TrashNet and proposes the optimized DenseNet121. The results show that the optimized DenseNet121 achieved the highest accuracy of 99.6%, when compared with other studies' CNNs. The data augmentation could perform higher efficiency on accuracy improvement of image classification than optimizing fully-connected-layer of DenseNet121 for TrashNet. To replace the function of the original classifier of DenseNet121 with fully-connected-layer can improve DenseNet121's performance. The optimized DenseNet121 further improved the accuracy and demonstrated the efficiency of using GA to optimize the neuron number and the dropout rate of fully-connected-layer. Gradient-weighted class activation mapping helped highlight the coarse features of the waste image and provide additional insight into the explainability of optimized DenseNet121.
作者
我是这篇论文的作者
点击您的名字以认领此论文并将其添加到您的个人资料中。
推荐
暂无数据