Lightweight Recyclable Garbage Detection Method Incorporating Attention Mechanism

GUO Zhou, HUANG Shi-hao, XIE Wen-ming, LYU Hui, ZHANG Xuan-xuan, CHEN Zhe

Packaging Engineering ›› 2023 ›› Issue (9) : 243-253.

PDF(3919 KB)
PDF(3919 KB)
Packaging Engineering ›› 2023 ›› Issue (9) : 243-253. DOI: 10.19554/j.cnki.1001-3563.2023.09.030

Lightweight Recyclable Garbage Detection Method Incorporating Attention Mechanism

  • GUO Zhou1, HUANG Shi-hao1, LYU Hui1, ZHANG Xuan-xuan1, CHEN Zhe1, XIE Wen-ming2
Author information +
History +

Abstract

The work aims to propose a lightweight method based on YOLOv4 to detect recyclable garbage, so as to address the problems of slow detection speed and large model weight files in the current garbage detection methods used by smart garbage sorting devices. The MobileNetV2 lightweight network was used as the backbone network of YOLOv4 and the depth-separable convolution was used to optimize the neck and head networks to reduce the parameters and computation to accelerate detection. The CBAM attention module was incorporated into the neck network to improve the sensitivity of the model to the target feature information. The K-means algorithm was used to re-cluster to get suitable self-built recyclable data with a priori frame for focused detection of targets. The experimental results showed that:the parameters were reduced to 17.0% of the original YOLOv4 model. The detected mAP reached 96.78%. The model weight file size was 46.6 MB, which was about 19.1% of the YOLOv4 model weight file. The detection speed was 20.46 frames/s, which was improved by 25.4%. Both the detection accuracy and the detection speed met the real-time detection requirements. The improved YOLOv4 model can guarantee high detection accuracy and good real-time performance in detection of recyclable garbage.

Cite this article

Download Citations
GUO Zhou, HUANG Shi-hao, XIE Wen-ming, LYU Hui, ZHANG Xuan-xuan, CHEN Zhe. Lightweight Recyclable Garbage Detection Method Incorporating Attention Mechanism[J]. Packaging Engineering. 2023(9): 243-253 https://doi.org/10.19554/j.cnki.1001-3563.2023.09.030
PDF(3919 KB)

Accesses

Citation

Detail

Sections
Recommended

/