A Hybrid Deep Learning Model for Trash Classification Based on Deep Trasnsfer Learning
Trash classification is an effective measure to protect the ecological environment and improve resource utilization. With the development of deep learning, it is possible to use the deep convolutional neural network for trash classification. In order to classify the trash of the TrashNet dataset, wh...
Saved in:
| Main Authors: | , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
Wiley
2022-01-01
|
| Series: | Journal of Electrical and Computer Engineering |
| Online Access: | http://dx.doi.org/10.1155/2022/7608794 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1832551044383506432 |
|---|---|
| author | Zhen Yuan Jinfeng Liu |
| author_facet | Zhen Yuan Jinfeng Liu |
| author_sort | Zhen Yuan |
| collection | DOAJ |
| description | Trash classification is an effective measure to protect the ecological environment and improve resource utilization. With the development of deep learning, it is possible to use the deep convolutional neural network for trash classification. In order to classify the trash of the TrashNet dataset, which consists of six classes of garbage images, this paper proposes a hybrid deep learning model based on deep transfer learning, which includes upper and lower streams. Firstly, the upper stream divides the input garbage image into category MPP (metal, paper, and plastic class) or category CGT (cardboard, glass, and trash class). Then, the lower stream predicts the exact class of trash according to the results of the upper stream. The proposed hybrid deep learning model achieves the best result with 98.5 % than that of the state-of-the-art approaches. Through the verification of CAM (class activation map), the proposed model can reasonably use the features of the image for classification, which explains the reason for the superior performance of this model. |
| format | Article |
| id | doaj-art-4b74d63b88d1444a9e148976e0917937 |
| institution | Kabale University |
| issn | 2090-0155 |
| language | English |
| publishDate | 2022-01-01 |
| publisher | Wiley |
| record_format | Article |
| series | Journal of Electrical and Computer Engineering |
| spelling | doaj-art-4b74d63b88d1444a9e148976e09179372025-02-03T06:05:01ZengWileyJournal of Electrical and Computer Engineering2090-01552022-01-01202210.1155/2022/7608794A Hybrid Deep Learning Model for Trash Classification Based on Deep Trasnsfer LearningZhen Yuan0Jinfeng Liu1School of Information EngineeringSchool of Information EngineeringTrash classification is an effective measure to protect the ecological environment and improve resource utilization. With the development of deep learning, it is possible to use the deep convolutional neural network for trash classification. In order to classify the trash of the TrashNet dataset, which consists of six classes of garbage images, this paper proposes a hybrid deep learning model based on deep transfer learning, which includes upper and lower streams. Firstly, the upper stream divides the input garbage image into category MPP (metal, paper, and plastic class) or category CGT (cardboard, glass, and trash class). Then, the lower stream predicts the exact class of trash according to the results of the upper stream. The proposed hybrid deep learning model achieves the best result with 98.5 % than that of the state-of-the-art approaches. Through the verification of CAM (class activation map), the proposed model can reasonably use the features of the image for classification, which explains the reason for the superior performance of this model.http://dx.doi.org/10.1155/2022/7608794 |
| spellingShingle | Zhen Yuan Jinfeng Liu A Hybrid Deep Learning Model for Trash Classification Based on Deep Trasnsfer Learning Journal of Electrical and Computer Engineering |
| title | A Hybrid Deep Learning Model for Trash Classification Based on Deep Trasnsfer Learning |
| title_full | A Hybrid Deep Learning Model for Trash Classification Based on Deep Trasnsfer Learning |
| title_fullStr | A Hybrid Deep Learning Model for Trash Classification Based on Deep Trasnsfer Learning |
| title_full_unstemmed | A Hybrid Deep Learning Model for Trash Classification Based on Deep Trasnsfer Learning |
| title_short | A Hybrid Deep Learning Model for Trash Classification Based on Deep Trasnsfer Learning |
| title_sort | hybrid deep learning model for trash classification based on deep trasnsfer learning |
| url | http://dx.doi.org/10.1155/2022/7608794 |
| work_keys_str_mv | AT zhenyuan ahybriddeeplearningmodelfortrashclassificationbasedondeeptrasnsferlearning AT jinfengliu ahybriddeeplearningmodelfortrashclassificationbasedondeeptrasnsferlearning AT zhenyuan hybriddeeplearningmodelfortrashclassificationbasedondeeptrasnsferlearning AT jinfengliu hybriddeeplearningmodelfortrashclassificationbasedondeeptrasnsferlearning |