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 条
  • [1] Multi-branch fusion network for hyperspectral image classification
    Gao, Hongmin
    Yang, Yao
    Lei, Sheng
    Li, Chenming
    Zhou, Hui
    Qu, Xiaoyu
    KNOWLEDGE-BASED SYSTEMS, 2019, 167 : 11 - 25
  • [2] Multi-Branch Spectral Channel Attention Network for Breast Cancer Histopathology Image Classification
    Cao, Lu
    Pan, Ke
    Ren, Yuan
    Lu, Ruidong
    Zhang, Jianxin
    ELECTRONICS, 2024, 13 (02)
  • [3] Multi-branch Aggregate Convolutional Neural Network for Image Classification
    Fan, Rui
    Jiang, Pinqun
    Zeng, Shangyou
    Li, Peng
    SERVICE-ORIENTED COMPUTING, ICSOC 2018, 2019, 11434 : 102 - 112
  • [4] Channel rearrangement multi-branch network for image super-resolution
    Wei, Deyun
    Wang, Zhaowu
    DIGITAL SIGNAL PROCESSING, 2022, 120
  • [5] Channel Attention Multi-Branch Network for Fine-Grained Image Recognition
    Wang Binzhou
    Xiao Zhiyong
    LASER & OPTOELECTRONICS PROGRESS, 2021, 58 (22)
  • [6] Multi-Branch-CNN: Classification of ion channel interacting peptides using multi-branch convolutional neural network
    Yan, Jielu
    Zhang, Bob
    Zhou, Mingliang
    Kwok, Hang Fai
    Siu, Shirley W., I
    COMPUTERS IN BIOLOGY AND MEDICINE, 2022, 147
  • [7] Hyperspectral image classification based on faster residual multi-branch spiking neural network
    Li, Yahui
    Liu, Yang
    Li, Rui
    Zhou, Liming
    Dang, Lanxue
    Mu, Huiyu
    Ge, Qiang
    COMPUTERS & GEOSCIENCES, 2025, 197
  • [8] Pruning Multi-Scale Multi-Branch Network for Small-Sample Hyperspectral Image Classification
    Bai, Yu
    Xu, Meng
    Zhang, Lili
    Liu, Yuxuan
    ELECTRONICS, 2023, 12 (03)
  • [9] A novel multi-branch hybrid neural network for motor imagery EEG signal classification
    Ma, Weifeng
    Xue, Haojie
    Sun, Xiaoyong
    Mao, Sijia
    Wang, Liudi
    Liu, Yang
    Wang, Yuchen
    Lin, Xuefen
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2022, 77
  • [10] A novel multi-branch wavelet neural network for sparse representation based object classification
    Nguyen, Tan-Sy
    Luong, Marie
    Kaaniche, Mounir
    Ngo, Long H.
    Beghdadi, Azeddine
    PATTERN RECOGNITION, 2023, 135