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] Classification of intervertebral disc using novel multi-branch convolutional residual network model
    Ram, Indradeo
    Kumar, Sanjay
    Keshri, Anup Kumar
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2025, 102
  • [12] A Multi-Branch Defogging Network Based on Fog Concentration Classification and Dark and Bright Channel Priors
    Zhang Q.
    Chi J.
    Chen Y.
    Zhang C.
    Jisuanji Yanjiu yu Fazhan/Computer Research and Development, 2024, 61 (03): : 762 - 779
  • [13] 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
  • [14] 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
  • [15] Remote Sensing Image Scene Classification Based on Deep Multi-branch Feature Fusion Network
    Zhang Tong
    Zheng En-rang
    Shen Jun-ge
    Gao An-tong
    ACTA PHOTONICA SINICA, 2020, 49 (05)
  • [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] Leadwise clustering multi-branch network for multi-label ECG classification
    Zhou, Feiyan
    Chen, Lingzhi
    MEDICAL ENGINEERING & PHYSICS, 2024, 130
  • [18] Improving Hyperspectral Image Classification with Compact Multi-Branch Deep Learning
    Islam, Md. Rashedul
    Islam, Md. Touhid
    Uddin, Md Palash
    Ulhaq, Anwaar
    REMOTE SENSING, 2024, 16 (12)
  • [19] Multi-branch CNN and grouping cascade attention for medical image classification
    Liu, Shiwei
    Yue, Wenwen
    Guo, Zhiqing
    Wang, Liejun
    SCIENTIFIC REPORTS, 2024, 14 (01):
  • [20] Classification of healthy, Alzheimer and Parkinson populations with a multi-branch neural network
    Francisco Pedrero-Sanchez, Jose
    Belda-Lois, Juan-Manuel
    Serra-Ano, Pilar
    Ingles, Marta
    Lopez-Pascual, Juan
    BIOMEDICAL SIGNAL PROCESSING AND CONTROL, 2022, 75