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 条
  • [11] Multi-Scale and Multi-Branch Convolutional Neural Network for Retinal Image Segmentation
    Jiang, Yun
    Liu, Wenhuan
    Wu, Chao
    Yao, Huixiao
    SYMMETRY-BASEL, 2021, 13 (03): : 1 - 25
  • [12] Multi-branch sustainable convolutional neural network for disease classification
    Naz, Maria
    Shah, Munam Ali
    Khattak, Hasan Ali
    Wahid, Abdul
    Asghar, Muhammad Nabeel
    Rauf, Hafiz Tayyab
    Khan, Muhammad Attique
    Ameer, Zoobia
    INTERNATIONAL JOURNAL OF IMAGING SYSTEMS AND TECHNOLOGY, 2023, 33 (05) : 1621 - 1633
  • [13] Multi-Branch Regression Network For Building Classification Using Remote Sensing Images
    Gui, Yuanyuan
    Li, Xiang
    Li, Wei
    Yue, Anzhi
    2018 10TH IAPR WORKSHOP ON PATTERN RECOGNITION IN REMOTE SENSING (PRRS), 2018,
  • [14] Seizure Types Classification Based on Multi-branch Hybrid Deep Learning Network
    Jia, Qingwei
    Liu, Jin-Xing
    Shang, Junling
    Dai, Lingyun
    Wang, Yuxia
    Hu, Wenrong
    Yuan, Shasha
    ADVANCED INTELLIGENT COMPUTING TECHNOLOGY AND APPLICATIONS, PT IV, ICIC 2024, 2024, 14865 : 462 - 474
  • [15] Scene Classification via Learning a Multi-Branch Convolutional Network
    Bian, Xiaoyong
    Chen, Chunfang
    Chen, Yang
    Fei, Xiongjun
    Tang, Jingshan
    2019 IEEE INTERNATIONAL CONFERENCE ON SYSTEMS, MAN AND CYBERNETICS (SMC), 2019, : 2555 - 2559
  • [16] MultiTrans: Multi-branch transformer network for medical image segmentation
    Zhang, Yanhua
    Balestra, Gabriella
    Zhang, Ke
    Wang, Jingyu
    Rosati, Samanta
    Giannini, Valentina
    COMPUTER METHODS AND PROGRAMS IN BIOMEDICINE, 2024, 254
  • [17] T-S Fuzzy Model Based Multi-Branch Deep Network Architecture
    Wang, Faguang
    Wang, Yue
    Wang, Hongmei
    Tang, Chaogang
    IEEE ACCESS, 2020, 8 : 155039 - 155046
  • [18] Multi-Branch Enhanced Discriminative Network for Vehicle Re-Identification
    Lian, Jiawei
    Wang, Da-Han
    Wu, Yun
    Zhu, Shunzhi
    IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2024, 25 (02) : 1263 - 1274
  • [19] Multi-Branch Cascade Receptive Field Residual Network
    Zhang, Xudong
    Liu, Wenjie
    Wu, Guoqing
    IEEE ACCESS, 2023, 11 : 82613 - 82623
  • [20] Channel Expansion Convolutional Network for Image Classification
    Yang, Yadong
    Wang, Xiaofeng
    Sun, Bowen
    Zhao, Quan
    IEEE ACCESS, 2020, 8 : 178414 - 178424