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

被引:44
作者
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 条
  • [21] MB-SCRSR: A Multi-branch Image Super-Resolution Network Based on Spatial and Channel Reconstruction
    Zhang, Panpan
    Yang, Xin
    Xia, Tingyu
    CIRCUITS SYSTEMS AND SIGNAL PROCESSING, 2025,
  • [22] Deep Convolutional Neural Network for Segmentation and Classification of Structural Multi-branch Cracks
    Kandula, Himavanth
    Koduri, Hrushith Ram
    Kalapatapu, Prafulla
    Pasupuleti, Venkata Dilip Kumar
    EUROPEAN WORKSHOP ON STRUCTURAL HEALTH MONITORING (EWSHM 2022), VOL 2, 2023, : 177 - 185
  • [23] Wireless Interference Classification with Low Complexity Multi-Branch Networks
    Ma, Song
    Cheng, Yufan
    Mou, Ying
    Wang, Pengyu
    Peng, Qihang
    Wang, Jun
    CHINA COMMUNICATIONS, 2023, 20 (04) : 382 - 394
  • [24] Remote Sensing Scene Classification via Multi-Branch Local Attention Network
    Chen, Si-Bao
    Wei, Qing-Song
    Wang, Wen-Zhong
    Tang, Jin
    Luo, Bin
    Wang, Zu-Yuan
    IEEE TRANSACTIONS ON IMAGE PROCESSING, 2022, 31 : 99 - 109
  • [25] Residual multi-branch distillation network for efficient image super-resolution
    Gao, Xiang
    Zhou, Ying
    Wu, Sining
    Wu, Xinrong
    Wang, Fan
    Hu, Xiaopeng
    MULTIMEDIA TOOLS AND APPLICATIONS, 2024, 83 (30) : 75217 - 75241
  • [26] MBNet: Multi-Branch Network for Extraction of Rural Homesteads Based on Aerial Images
    Wei, Ren
    Fan, Beilei
    Wang, Yuting
    Zhou, Ailian
    Zhao, Zijuan
    REMOTE SENSING, 2022, 14 (10)
  • [27] MMCNN: A Multi-branch Multi-scale Convolutional Neural Network for Motor Imagery Classification
    Jia, Ziyu
    Lin, Youfang
    Wang, Jing
    Yang, Kaixin
    Liu, Tianhang
    Zhang, Xinwang
    MACHINE LEARNING AND KNOWLEDGE DISCOVERY IN DATABASES, ECML PKDD 2020, PT III, 2021, 12459 : 736 - 751
  • [28] Multiclass classification of motor imagery tasks based on multi-branch convolutional neural network and temporal convolutional network model
    Yu, Shiqi
    Wang, Zedong
    Wang, Fei
    Chen, Kai
    Yao, Dezhong
    Xu, Peng
    Zhang, Yong
    Wang, Hesong
    Zhang, Tao
    CEREBRAL CORTEX, 2024, 34 (02)
  • [29] Multi-dimensional, multi-branch hyperspectral remote sensing image classification with limited training samples
    Zeng, Yiliang
    Lv, Zhiwu
    Zhang, Hao
    Zhao, Jiahong
    SIGNAL IMAGE AND VIDEO PROCESSING, 2024, 18 (10) : 7199 - 7210
  • [30] Classification and segmentation of defect carrots using deep multi-branch models fusion network
    Xie W.
    Wei S.
    Zheng Z.
    Yang G.
    Ding X.
    Yang D.
    Nongye Gongcheng Xuebao/Transactions of the Chinese Society of Agricultural Engineering, 2021, 37 (02): : 177 - 186