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 条
  • [31] Medical image segmentation with an emphasis on prior convolution and channel multi-branch attention
    Wang, Yuenan
    Wang, Hua
    Zhang, Fan
    DIGITAL SIGNAL PROCESSING, 2025, 162
  • [32] 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
  • [33] A deep multi-branch attention model for histopathological breast cancer image classification
    Ding, Rui
    Zhou, Xiaoping
    Tan, Dayu
    Su, Yansen
    Jiang, Chao
    Yu, Guo
    Zheng, Chunhou
    COMPLEX & INTELLIGENT SYSTEMS, 2024, 10 (03) : 4571 - 4587
  • [34] A Multi-branch Feature Fusion Model Based on Convolutional Neural Network for Hyperspectral Remote Sensing Image Classification
    Zhang, Jinli
    Chen, Ziqiang
    Ji, Yuanfa
    Sun, Xiyan
    Bai, Yang
    INTERNATIONAL JOURNAL OF ADVANCED COMPUTER SCIENCE AND APPLICATIONS, 2023, 14 (06) : 147 - 156
  • [35] Multi-branch Recurrent Attention Convolutional Neural Network with Evidence Theory for Fine-Grained Image Classification
    Xu, Zhikang
    Zhang, Bofeng
    Fu, Haijie
    Yue, Xiaodong
    Lv, Ying
    BELIEF FUNCTIONS: THEORY AND APPLICATIONS (BELIEF 2021), 2021, 12915 : 177 - 184
  • [36] An efficient medical image classification network based on multi-branch CNN, token grouping Transformer and mixer MLP
    Liu, Shiwei
    Wang, Liejun
    Yue, Wenwen
    APPLIED SOFT COMPUTING, 2024, 153
  • [37] Multi-Branch Hybrid Network Based on Adaptive Selection of Spatial-Spectral Kernel for Hyperspectral Image Classification
    Wang, Cailing
    Fu, He
    Wang, Hongwei
    IEEE ACCESS, 2023, 11 : 80503 - 80517
  • [38] 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
  • [39] 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
  • [40] 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,