Road Trash Detection Algorithm of RT-DETR Integrated with State Space

As global urbanization progresses rapidly, the issue of urban road trash is becoming increasingly severe. Traditional manual cleaning methods are inefficient and struggle to meet the growing demand. This paper proposes an improved road trash detection algorithm based on RT-DETR (Real-Time Detection Transformer) and State Space Model (SSM) to enhance detection accuracy and robustness. The algorithm incorporates an innovative feature pyramid structure and a state-space-based internal scale feature enhancement module, effectively addressing object detection in complex scenes as well as challenges associated with occlusions and small object detection. We constructed a diverse urban road trash detection dataset comprising over 60,000 images, providing a solid foundation for the training and testing of the algorithm. Experimental results show that the proposed model improves the mean average precision (mAP) on the COCO dataset by 0.8 percentage points and achieves a 1.2 percentage point improvement on our custom road trash detection dataset, with a 2.1 percentage point increase in the precision of small object detection. The proposed algorithm achieves a favorable balance between real-time performance and accuracy, significantly enhancing trash detection in complex environments, particularly in terms of small object recognition and detection accuracy under occlusion conditions.

Authors

Institutions

Publication Details

Published
2024-11-22
DOI
https://doi.org/10.1109/icicml63543.2024.10958111
Primary Topic
Advanced Algorithms and Applications
Type
article
Field-Weighted Citation Impact
0.00
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
article

Road Trash Detection Algorithm of RT-DETR Integrated with State Space

Guoqing Yang, Yeqin Shao, Xinyong Zhao, Zexing Luo
Advanced Algorithms and Applications
article

Road Trash Detection Algorithm of RT-DETR Integrated with State Space

Guoqing Yang, Yeqin Shao, Xinyong Zhao, Zexing Luo
article en

Abstract

As global urbanization progresses rapidly, the issue of urban road trash is becoming increasingly severe. Traditional manual cleaning methods are inefficient and struggle to meet the growing demand. This paper proposes an improved road trash detection algorithm based on RT-DETR (Real-Time Detection Transformer) and State Space Model (SSM) to enhance detection accuracy and robustness. The algorithm incorporates an innovative feature pyramid structure and a state-space-based internal scale feature enhancement module, effectively addressing object detection in complex scenes as well as challenges associated with occlusions and small object detection. We constructed a diverse urban road trash detection dataset comprising over 60,000 images, providing a solid foundation for the training and testing of the algorithm. Experimental results show that the proposed model improves the mean average precision (mAP) on the COCO dataset by 0.8 percentage points and achieves a 1.2 percentage point improvement on our custom road trash detection dataset, with a 2.1 percentage point increase in the precision of small object detection. The proposed algorithm achieves a favorable balance between real-time performance and accuracy, significantly enhancing trash detection in complex environments, particularly in terms of small object recognition and detection accuracy under occlusion conditions.

Nantong University (CN), Suzhou Research Institute (CN), Shanghai Huayi Group (China) (CN), Zhejiang University (CN)
Sustainable cities and communities
Openalex Percentile: Top 63%
Advanced Algorithms and Applications
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.