Frontiers of Optoelectronics, Volume. 15, Issue 2, 12200(2022)

A similarity-guided segmentation model for garbage detection under road scene

Caiyun Zheng, Danhua Cao, and Cheng Hu1
Author Affiliations
  • School of Optical and Electronic Information, Huazhong University of Science and Technology, Wuhan 430074, China
  • show less

    The development of computer vision technology provides a possible path for realizing intelligent control of road sweepers to reduce energy waste in urban street cleaning work. For garbage segmentation of seven categories under road scene, we introduce an efficient deep-learning-based method. Our model follows a lightweight structure with a feature pyramid attention (FPA) module employed in the decoder to enhance feature integration at multi-levels. Besides, a similarity guidance (SG) module is added to the decoder branches, which calculates the cosine distance between learned prototypes and feature maps to guide the segmentation results from a metric learning perspective. Our model has less than 3 M parameters and can run at over 65 FPS in an RTX 2070 GPU. Experimental results demonstrate that our method can yield competitive results in terms of speed and accuracy trade-off, with overall mean intersection-over-union (mIoU) reaching 0.87 and 0.67, respectively, on two garbage data sets we built. Besides, our model can perform acceptable category-balanced segmentation from less than 20 annotated images per category by introducing the SG module.

    Tools

    Get Citation

    Copy Citation Text

    Caiyun Zheng, Danhua Cao, Cheng Hu1. A similarity-guided segmentation model for garbage detection under road scene[J]. Frontiers of Optoelectronics, 2022, 15(2): 12200

    Download Citation

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

    Category: RESEARCH ARTICLE

    Received: Feb. 2, 2021

    Accepted: Jul. 5, 2021

    Published Online: Jan. 18, 2023

    The Author Email:

    DOI:10.1007/s12200-022-00004-9

    Topics