Zone-YOLO: Vision-Language Object Detection Using Zone Prompt
Object detection in complex traffic scenarios is crucial for Intelligent Transportation Systems (ITS). At present, most real-time traffic object detection methods primarily rely on YOLO-style vision-only detectors, limiting their potential for further improvement. Vision-Language Object Detection (VLOD) has made promising progress currently, yet its adoption in the realm of ITS remains limited. Previous VLOD methods utilize text features in the classification task, without fully exploring their impact on the regression process for object localization. Besides, existing multi-modal fusion approaches fail to fuse text features with multi-scale image features at corresponding scales, which is detrimental to the representation capability of the model. In this work, we dive into the limitations above and introduce Zone-YOLO to improve the VLOD to a new level. Specifically, we propose Scale-Aware Modal Fusion (SAMF) to fully exploit the text and image features and learn to fuse the multi-modal representations seamlessly at different scales with channel- and modal-wise enhancement. Moreover, we present a novel Zone Prompt learning method to introduce text features into regression process and capture the zone-class-entity triple co-occurrence, which significantly improves the localization performance of the model. Extensive experiments show that Zone-YOLO outperforms the comparative methods by a considerable margin, achieving 55.1 AP, 72.1 AP50 and 71.2 APL on COCO. The competitive results on BDD100K and VisDrone2019 further demonstrate the superiority of Zone-YOLO on efficient traffic object detection.
Authors
- Rui Fan (ORCID: https://orcid.org/0000-0001-8768-932X)
- Jiaxiong Yang (ORCID: https://orcid.org/0000-0002-7499-2474)
- Weidong Zhao (ORCID: https://orcid.org/0000-0003-4164-5190)
- Yougang Sun (ORCID: https://orcid.org/0000-0002-1549-0108)
- Xianhui Liu (ORCID: https://orcid.org/0000-0002-5088-3628)
- Ning Jia (ORCID: https://orcid.org/0000-0002-2395-3432)
Institutions
- Tongji University (CN)
Publication Details
- Journal
- IEEE Transactions on Intelligent Transportation Systems
- Published
- 2024-12-16
- DOI
- https://doi.org/10.1109/tits.2024.3510117
- Citations
- 3
- Primary Topic
- Advanced Image and Video Retrieval Techniques
- Type
- article
- Field-Weighted Citation Impact
- 0.75
Funders
- National Natural Science Foundation of China