Machine learning;
Image classification;
Classification algorithms;
Prediction algorithms;
Neural networks;
Support vector machines;
Training;
Garbage image classification;
deep learning;
feature information fusion;
multi-branch;
small data sets;
CONVOLUTIONAL NEURAL-NETWORK;
ENVIRONMENTAL-POLLUTION;
MODEL;
RECOGNITION;
SALIENT;
D O I:
10.1109/ACCESS.2020.3016116
中图分类号:
TP [自动化技术、计算机技术];
学科分类号:
0812 ;
摘要:
Due to the lack of data available for training, deep learning hardly performed well in the field of garbage image classification. We choose the TrashNet data set which is widely used in the field of garbage image classification, and try to overcome data deficiencies in this field by optimizing the network structure. In this article, it is found that the deeper network and short-circuit connection, which are generally accepted in the field of deep learning, will not work well on the TrashNet data set. By analyzing and modifying the network structure, we propose an effective method to improve the network performance on TrashNet data set. This method widens the network by expanding branches, and then uses add layers to realize the fusion of feature information. It can make full use of feature information at slight additional computational cost. Using this method to replace the core structure of the Xception network, the performance of the improved network has been improved greatly. Finally, the M-b Xception network proposed by us achieves 94.34% classification accuracy on the TrashNet data set, and has certain advantages over some state-of-the-art methods on multiple indicators. The python code can be download from https://github.com/scp19801980/Trash-classify-M_b-Xception.
机构:
Univ Macau, Dept Comp & Informat Sci, PAMI Res Grp, Taipa, Macao, Peoples R ChinaUniv Macau, Dept Comp & Informat Sci, PAMI Res Grp, Taipa, Macao, Peoples R China
Yan, Jielu
Zhang, Bob
论文数: 0引用数: 0
h-index: 0
机构:
Univ Macau, Dept Comp & Informat Sci, PAMI Res Grp, Taipa, Macao, Peoples R ChinaUniv Macau, Dept Comp & Informat Sci, PAMI Res Grp, Taipa, Macao, Peoples R China
Zhang, Bob
Zhou, Mingliang
论文数: 0引用数: 0
h-index: 0
机构:
Chongqing Univ, Sch Comp Sci, Chongqing, Peoples R ChinaUniv Macau, Dept Comp & Informat Sci, PAMI Res Grp, Taipa, Macao, Peoples R China
Zhou, Mingliang
Kwok, Hang Fai
论文数: 0引用数: 0
h-index: 0
机构:
Univ Macau, Fac Hlth Sci, Dept Biomed Sci, Taipa, Macao, Peoples R ChinaUniv Macau, Dept Comp & Informat Sci, PAMI Res Grp, Taipa, Macao, Peoples R China
Kwok, Hang Fai
Siu, Shirley W., I
论文数: 0引用数: 0
h-index: 0
机构:
Univ Macau, Dept Comp & Informat Sci, Taipa, Macao, Peoples R China
Univ St Joseph, Inst Sci & Environm, Estr Marginal da Ilha Verde, Taipa, Macao, Peoples R ChinaUniv Macau, Dept Comp & Informat Sci, PAMI Res Grp, Taipa, Macao, Peoples R China