@inproceedings{
wang2024uniyolo,
title={Uni-{YOLO}: Vision-Language Model-Guided {YOLO} for Robust and Fast Universal Detection in the Open World},
author={Xudong Wang and Weihong Ren and Xi'ai Chen and Huijie Fan and Yandong Tang and Zhi Han},
booktitle={ACM Multimedia 2024},
year={2024},
url={https://openreview.net/forum?id=iUrEqR14Au}
}
Universal object detectors aim to detect any object in any scene without human annotation, exhibiting superior generalization. However, the current universal object detectors show degraded performance in harsh weather, and their insufficient real-time capabilities limit their application. In this paper, we present Uni-YOLO, a universal detector designed for complex scenes with real-time performance. Uni-YOLO is a one-stage object detector that uses general object confidence to distinguish between objects and backgrounds, and employs a grid cell regression method for real-time detection. To improve its robustness in harsh weather conditions, the input of Uni-YOLO is adaptively enhanced with a physical model-based enhancement module. During training and inference, Uni-YOLO is guided by the extensive knowledge of the vision-language model CLIP. An object augmentation method is proposed to improve generalization in training by utilizing multiple source datasets with heterogeneous annotations. Furthermore, an online self-enhancement method is proposed to allow Uni-YOLO to further focus on specific objects through self-supervised fine-tuning in a given scene. Extensive experiments on public benchmarks and a UAV deployment are conducted to validate its superiority and practical value.
Uni-YOLO: Vision-Language Model-Guided YOLO for Robust and Fast Universal Detection in the Open World
Xudong Wang Weihong Ren Xi'ai Chen Huijie Fan Yandong Tang Zhi Han