ABSTRACT
When dealing with garbage image classification task, with the deepening of network layer, convolutional neural network will lead to gradient disappearance / explosion and large time consumption. Therefore, a garbage image classification method combining improved convolutional neural network and long short-term memory network is proposed. Taking ResNet-50 as the network backbone, it is optimized by using deep separable convolution and attention mechanism. At the same time, supplemented by LSTM, the features extracted by convolution network and cyclic network are fused to complete classification output. Ablation experiments are carried out on this model and compared with other typical convolutional neural networks. The results show that the accuracy of this model increases by an average of 4.5%. The introduction of deep separable convolution can reduce the training time by about 35.4% compared with the baseline method.
- Bureau of statistics of the people's Republic of China. China Statistical Yearbook [M]. Beijing: China Statistics Press, 2020. (in Chinese).Google Scholar
- Yue Xiaohua.Improve the local laws and regulations system of domestic waste classification [N]. China Environment News, 2021-08-16(003).(in Chinese).Google Scholar
- KRIZHEVSKY A, SUTSKEVER I, HINTON G. ImageNet classification with deep convolutional neural networks [C]//International Conference on Neural Information Processing Systems. Lake Tahoe: Curran Associates Inc, 2012: 1097-1105.Google Scholar
- SZEGEDY C, LIU W, JIA Y, et al. Going deeper with convolutions [C]// Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition. Boston: IEEE,2015: 1-9.Google Scholar
- SIMONYAN K, ZISSERMAN A. Very deep convolutional networks for large-scale image recognition [EB/OL]. [2021-02-22]. http://arxiv.org/abs/1409.1556.Google Scholar
- HE K, ZHANG X, REN S, Deep residual learning for image recognition [C]// 2016 the IEEE Conference on Computer Vision and Pattern Recognition. Las Vegas: IEEE, 2016:770-778.Google Scholar
- Kang Zhuang, Yang Jie, Guo Haoqi. Design of automatic garbage classification system based on machine vision [J]. Journal of Zhejiang University (Engineering Edition),2020, 54(7): 1272-1280. (in Chinese).Google Scholar
- GAO Ming, CHEN Yuhan, ZHANG Zehui, Garbage image classification algorithm based on new spatial attention mechanism and transfer learning[J]. System Engineering Theory and Practice, 2021, 41(2): 498-512. (in Chinese).Google Scholar
- Ying Liu,Zhishan Ge,Guoyun Lv,Shikai Wang. Research on Automatic Garbage Detection System Based on Deep Learning and Narrowband Internet of Things[J].Journal of Physics: Conference Series,2018,1069(1):012032.Google Scholar
- Bircanoglu C, Atay M, Be ş er F, et al. Recyclenet: Intelligent waste sorting using deep neural networks[C]// 2018 Innovations in Intelligent Systems and Applications (INISTA). IEEE, 2018: 1-7.Google Scholar
- Mao W L, Chen W C, Wang C T, et al. Recycling waste classification using optimized convolutional neural network[J]. Resources, Conservation and Recycling, 2021, 164: 105132.Google ScholarCross Ref
- Adedeji O,Wang Z. Intelligent waste classification system using deep learning convolutional neural network[J]. ProcediaManufacturing, 2019, 35: 607-612.Google Scholar
- Song F, Zhang Y, Zhang J. Optimization of CNN-based Garbage Classification Model[C]// Proceedings of the 4th International Conference on Computer Science and Application Engineering. 2020: 1-5.Google Scholar
- Yuan Jianye, Nan Xinyuan, Cai Xin, Li Chengrong. Garbage image classification method by lightweight residual network[J].Environmental Engineering,2021,39(02):110-115.(in Chinese).Google Scholar
- Pan lie, Zeng Cheng, Zhang Haifeng, Text emotion analysis method combining generalized autoregressive pre training language model and cyclic convolution neural network [J/OL]. Computer application:1-9[2021-09-15].http://kns.cnki.net/kcms/detail/51.1307.TP.20210906.1654.002.html.(in Chinese).Google Scholar
- Fan Hao, Li Pengfei. Emotion analysis of short text based on FastText word vector and bidirectional GRU recurrent neural network -Taking microblog comment text as an example [J]. Information science, 2021,39(04):15-22.(in Chinese).Google Scholar
- Li Jun, Li Ming. Research on facial expression recognition based on multi scale CNN and Bi- LSTM [J]. Journal of Beijing Union University,2021,35(01):35-39+44.(in Chinese).Google Scholar
- Liang Lianhui, Li Jun, Zhang Shaoquan. Hyperspectral image classification method based on 3D octave convolution and Bi -RNN attention network [J/OL]. Acta PHOTONICA Sinica:1-13[2021-09-15].http://kns.cnki.net/kcms/detail/61.1235.O4.20210831.1701.058.html.(in Chinese).Google Scholar
- Hochreiter S, Schmidhuber J. Long short-term memory.Neural Computation, 1997, 9(8): 1735–1780. [doi: 10.1162/neco.1997.9.8.1735]Google ScholarDigital Library
- WOO S, PARK J, LEE J Y, CBAM: convolutional block attention module [C]// European Conference on Computer Vision. Munich: Springer, 2018: 3–19.Google Scholar
- CHOLLET, François. Xception: Deep learning with depthwise separable convolutions[C]//Proceedings of the IEEE conference on computer vision and pattern recognition. 2017:1251-1258.Google Scholar
Index Terms
- Garbage image classification method based on improved convolution neural network and long short-term memory network
Recommendations
Aviation visibility forecasting by integrating Convolutional Neural Network and long short-term memory network
Weather forecasts are essential to aviation safety. Unreliable forecasts not only cause problems to pilots and air traffic controllers, but also lead to aviation accidents and incidents. To enhance the forecast accuracy, an integrated model comprising a ...
Long-Short-Term Memory Based on Adaptive Convolutional Network for Time Series Classification
AbstractDeep learning technology is effective to solve time series classification tasks. The existing deep learning algorithms with fixed step convolution cannot effectively extract and focus on multi-scale features. Based on the complexity and long-term ...
ResNext50 based convolution neural network-long short term memory model for plant disease classification
AbstractAgricultural problems need to be dealt with advanced computing methods to increase food productivity. Automatic classification of plant disease using deep learning methods helps to analyze the food quality and productivity. The existing methods ...
Comments