A Novel Multi-Branch Channel Expansion Network for Garbage Image Classification

被引:45
作者
Shi, Cuiping [1 ,2 ]
Xia, Ruiyang [1 ]
Wang, Liguo [2 ]
机构
[1] Qiqihar Univ, Coll Commun & Elect Engn, Qiqihar 161000, Peoples R China
[2] Harbin Engn Univ, Coll Informat & Commun Engn, Harbin 150001, Peoples R China
基金
中国博士后科学基金; 中国国家自然科学基金;
关键词
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.
引用
收藏
页码:154436 / 154452
页数:17
相关论文
共 50 条
  • [41] Multi-Branch Deep Learning Framework for Land Scene Classification in Satellite Imagery
    Khan, Sultan Daud
    Basalamah, Saleh
    [J]. REMOTE SENSING, 2023, 15 (13)
  • [42] A Multi-Branch Residual Network Based on Depth Correlation Features for the Classification of Chinese Ink Paintings
    Dong, Qingshuang
    Zhai, Yanbo
    Zhu, Jianfeng
    Wang, Lei
    Wu, Bing
    [J]. INTERNATIONAL JOURNAL OF UNCERTAINTY FUZZINESS AND KNOWLEDGE-BASED SYSTEMS, 2024, 32 (07) : 1015 - 1035
  • [43] Highly Efficient Ultrawideband Power Amplifier Based on a Novel Multi-Branch Matching Network
    Xuan, Xuefei
    Cheng, Zhiqun
    Zhang, Zhiwei
    Le, Chao
    [J]. IEEE MICROWAVE AND WIRELESS TECHNOLOGY LETTERS, 2023, 33 (09): : 1325 - 1328
  • [44] A novel medical image segmentation approach by using multi-branch segmentation network based on local and global information synchronous learning
    Jin, Shangzhu
    Yu, Sheng
    Peng, Jun
    Wang, Hongyi
    Zhao, Yan
    [J]. SCIENTIFIC REPORTS, 2023, 13 (01)
  • [45] Efficient multi-branch dynamic fusion network for super-resolution of industrial component image
    Wang, Guanqiang
    Chen, Mingsong
    Lin, Y. C.
    Tan, Xianhua
    Zhang, Chizhou
    Yao, Wenxin
    Gao, Baihui
    Li, Kai
    Li, Zehao
    Zeng, Weidong
    [J]. DISPLAYS, 2024, 82
  • [46] INVARIANT FEATURE EXTRACTION FOR IMAGE CLASSIFICATION VIA MULTI-CHANNEL CONVOLUTIONAL NEURAL NETWORK
    Mei, Shaohui
    Jiang, Ruoqiao
    Ji, Jingyu
    Sun, Jun
    Peng, Yang
    Zhang, Yifan
    [J]. 2017 INTERNATIONAL SYMPOSIUM ON INTELLIGENT SIGNAL PROCESSING AND COMMUNICATION SYSTEMS (ISPACS 2017), 2017, : 491 - 495
  • [47] A feature enhancement network based on image partitioning in a multi-branch encoder-decoder architecture
    Wang, Yuefei
    Zhang, Yutong
    Zhang, Li
    Wan, Yuxuan
    Chen, Zhixuan
    Xu, Yuquan
    Cao, Ruixin
    Zhao, Liangyan
    Yang, Yixi
    Yu, Xi
    [J]. KNOWLEDGE-BASED SYSTEMS, 2025, 311
  • [48] Low-Light Image Enhancement Based on Multi-Branch All Convolutional Neural Network
    Wu Ruoyou
    Wang Dexing
    Yuan Hongchun
    Peng, Gong
    Chen Guanqi
    Dan, Wang
    [J]. LASER & OPTOELECTRONICS PROGRESS, 2020, 57 (14)
  • [49] Affine invariant feature matching of oblique images based on multi-branch network
    Zhang C.
    Yao G.
    Zhang L.
    Ai H.
    Man X.
    Huang P.
    [J]. Cehui Xuebao/Acta Geodaetica et Cartographica Sinica, 2021, 50 (05): : 641 - 651
  • [50] A Multi-Branch Feature Fusion Strategy Based on an Attention Mechanism for Remote Sensing Image Scene Classification
    Shi, Cuiping
    Zhao, Xin
    Wang, Liguo
    [J]. REMOTE SENSING, 2021, 13 (10)