Laser & Optoelectronics Progress, Volume. 60, Issue 22, 2210002(2023)

Outdoor Garbage Detection Based on Improved YOLOv5

Shengxuan Chen and Aimin Wang*
Author Affiliations
  • School of Instrument Science and Engineering, Southeast University, Nanjing 210096, Jiangsu, China
  • show less

    Owing to the increasing severity of garbage pollution, automatic garbage detection has become significantly important in practice. The detection mechanism of YOLOv5 is improved in this study to achieve better performance in outdoor garbage detection against a complicated background. Moreover, here, a garbage dataset is constructed comprising six garbage image types collected in a complex background; subsequently, a simple yet efficient method is proposed to generate ground truth heat maps of garbage objects presented in the images. We treat the corresponding heat maps as a quantization standard and then obtain a branch structure based on YOLOv5 by conducting experiments to generate predicted heat maps. Subsequently, the predicted heat maps are sent back to the backbone structure of YOLOv5 to increase the spatial attention weights of the feature maps in the training process to improve the performance of the entire target detection network. Only a few parameters are added to the improved network, which generates proper predicted heat maps and the performance of garbage detection has been greatly improved.

    Tools

    Get Citation

    Copy Citation Text

    Shengxuan Chen, Aimin Wang. Outdoor Garbage Detection Based on Improved YOLOv5[J]. Laser & Optoelectronics Progress, 2023, 60(22): 2210002

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Image Processing

    Received: Jan. 9, 2023

    Accepted: Feb. 22, 2023

    Published Online: Nov. 3, 2023

    The Author Email: Wang Aimin (wangam@seu.edu.cn)

    DOI:10.3788/LOP230472

    Topics