Garbage classification target detection model based on Yolo_ES
DOI:
CSTR:
Author:
Affiliation:

1.School of Electrical Engineering and Automation, Henan Polytechnic University,Jiaozuo 454000, China; 2.Henan Key Laboratory of Intelligent Detection and Control of Coal Mine Equipment, Henan Polytechnic University,Jiaozuo 454000, China

Clc Number:

TP391.4

Fund Project:

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    At present, garbage classification has been widely concerned by the government and society. Facing the demand for real-time and accurate judgment of waste classification in the sorting process, a Yolo_ES target detection algorithm was proposed. The algorithm takes Yolov4 as the basic network. Firstly, EfficientNet is used as the backbone feature extraction network to realize the lightweight of the algorithm; secondly, the MBConv module is reconstructed by the attention mechanism ECA to filter out the highquality information, enhance the feature extraction ability of the model and reduce the number of parameters; at the same time, aiming at the problem that it is easy to lose detailed information in the max-pooling, the SoftPool is used to replace the MaxPool layer in the SPP module to retain more fine-grained feature information. The experiments are conducted on the self-made HPU_WASTE garbage classification dataset, and the results show that compared with the Yolov4 basic network, Yolo_ES model increases the map from 91.81% to 96.06%, and the model size is compressed by 75.45%. Meanwhile, the processing time of each image is 58ms; Compared with other target detection networks, this model has better robustness and better detection performance.

    Reference
    Related
    Cited by
Get Citation
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:
  • Revised:
  • Adopted:
  • Online: March 11,2024
  • Published:
Article QR Code