A Novel Framework for Trash Classification Using Deep Transfer Learning

Nowadays, society is growing and crowded, the construction of automatic smart waste sorter machine utilizing the intelligent sensors is important and necessary. To build this system, trash classification from trash images is an important issue in computer vision to be addressed for integrating into sensors. Therefore, this study proposes a robust model using deep neural networks to classify trash automatically which can be applied in smart waste sorter machines. Firstly, we collect the VN-trash dataset that consists of 5904 images belonging to three different classes including Organic, Inorganic and Medical wastes from Vietnam. Next, this study develops a deep neural network model for trash classification named DNN-TC which is an improvement of ResNext model to improve the predictive performance. Finally, the experiments are conducted to compare the performances of DNN-TC and the state-of-the-art methods for trash classification on VN-trash dataset as well as Trashnet dataset to show the effectiveness of the proposed model. The experimental results indicate that DNN-TC yields 94% and 98% in terms of accuracy for Trashnet and VN-trash datasets respectively and thus it outperforms the state-of-the-art methods for trash classification on both experimental datasets.

[1]  Ngoc Thanh Nguyen,et al.  A fast and accurate approach for bankruptcy forecasting using squared logistics loss with GPU-based extreme gradient boosting , 2019, Inf. Sci..

[2]  Geoffrey E. Hinton,et al.  ImageNet classification with deep convolutional neural networks , 2012, Commun. ACM.

[3]  Hyoung Il Son,et al.  Unmanned Aerial Vehicles in Agriculture: A Review of Perspective of Platform, Control, and Applications , 2019, IEEE Access.

[4]  Zhuowen Tu,et al.  Aggregated Residual Transformations for Deep Neural Networks , 2016, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[5]  Jeongrae Kim,et al.  Fraud detection for job placement using hierarchical clusters-based deep neural networks , 2019, Applied Intelligence.

[6]  Xiaogang Xiong,et al.  Multilayer Hybrid Deep-Learning Method for Waste Classification and Recycling , 2018, Comput. Intell. Neurosci..

[7]  Javad Rahebi,et al.  Face Recognition with Symmetrical Face Training Samples Based on Local Binary Patterns and the Gabor Filter , 2019, Symmetry.

[8]  Trang Nguyen,et al.  A hybrid framework for smile detection in class imbalance scenarios , 2019, Neural Computing and Applications.

[9]  Min Zuo,et al.  CropDeep: The Crop Vision Dataset for Deep-Learning-Based Classification and Detection in Precision Agriculture , 2019, Sensors.

[10]  Haibo Hu,et al.  Automated Lung Nodule Detection and Classification Using Deep Learning Combined with Multiple Strategies , 2019, Sensors.

[11]  Sung Wook Baik,et al.  Oversampling Techniques for Bankruptcy Prediction: Novel Features from a Transaction Dataset , 2018, Symmetry.

[12]  Cenk Bircanoglu,et al.  RecycleNet: Intelligent Waste Sorting Using Deep Neural Networks , 2018, 2018 Innovations in Intelligent Systems and Applications (INISTA).

[13]  Gary Thung,et al.  Classification of Trash for Recyclability Status , 2016 .

[14]  Antoanela Naaji,et al.  A Modified Deep Convolutional Neural Network for Abnormal Brain Image Classification , 2019, IEEE Access.

[15]  U. Rajendra Acharya,et al.  Application of deep convolutional neural network for automated detection of myocardial infarction using ECG signals , 2017, Inf. Sci..

[16]  Iwan Kurnianto Wibowo,et al.  Visual-based trash detection and classification system for smart trash bin robot , 2018, 2018 International Electronics Symposium on Knowledge Creation and Intelligent Computing (IES-KCIC).

[17]  Lei He,et al.  Facial Expression Recognition Based on Fusion Features of Center-Symmetric Local Signal Magnitude Pattern , 2019, IEEE Access.

[18]  Trang Nguyen,et al.  Race Recognition Using Deep Convolutional Neural Networks , 2018, Symmetry.

[19]  Andrew Zisserman,et al.  Very Deep Convolutional Networks for Large-Scale Image Recognition , 2014, ICLR.

[20]  Wanyi Li,et al.  Identification and Classification of Maize Drought Stress Using Deep Convolutional Neural Network , 2019, Symmetry.

[21]  Jian Sun,et al.  Deep Residual Learning for Image Recognition , 2015, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[22]  Jongwoo Lee,et al.  A study on novel filtering and relationship between input-features and target-vectors in a deep learning model for stock price prediction , 2018, Applied Intelligence.

[23]  Murat Haciomeroglu,et al.  Classification of TrashNet Dataset Based on Deep Learning Models , 2018, 2018 IEEE International Conference on Big Data (Big Data).

[24]  Sung Wook Baik,et al.  Improving Electric Energy Consumption Prediction Using CNN and Bi-LSTM , 2019, Applied Sciences.

[25]  U. Rajendra Acharya,et al.  Deep convolutional neural network for the automated diagnosis of congestive heart failure using ECG signals , 2018, Applied Intelligence.

[26]  Yun Zhang,et al.  Identification of Apple Leaf Diseases Based on Deep Convolutional Neural Networks , 2017, Symmetry.

[27]  Anjan Gudigar,et al.  Deep convolution neural network for accurate diagnosis of glaucoma using digital fundus images , 2018, Inf. Sci..

[28]  Bogdan Raducanu,et al.  Automatic Image-Based Waste Classification , 2019, IWINAC.

[29]  Frede Blaabjerg,et al.  A Review of Internet of Energy Based Building Energy Management Systems: Issues and Recommendations , 2018, IEEE Access.