arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2305.07528cs.CVcs.AI
收藏

WEDGE:由生成式视觉-语言模型构建的多天气自动驾驶数据集

WEDGE: A multi-weather autonomous driving dataset built from generative vision-language models

Carnegie Mellon University(卡内基梅隆大学);Symbiosis International University(共生国际大学);Symbiosis Centre for Applied AI(共生应用人工智能中心);Symbiosis Institute of Technology(共生技术学院)

Aboli Marathe, Deva Ramanan, Rahee Walambe, Ketan Kotecha

AI总结:

针对极端天气下自动驾驶感知性能下降的问题,提出基于生成式视觉-语言模型构建的WEDGE数据集,含3360张图像和16513个边界框,可提升检测器在真实天气基准上的SOTA性能。

摘要:

开放道路对自动驾驶感知提出了诸多挑战,包括极端天气条件导致的能见度低下。在良好天气数据集上训练的模型在检测这些分布外场景时经常失败。为了增强感知中的对抗鲁棒性,我们引入了WEDGE(通过DALL-E生成天气图像):一个使用视觉-语言生成模型通过提示生成的合成数据集。WEDGE包含16种极端天气条件下的3360张图像,并手动标注了16513个边界框,支持天气分类和2D目标检测任务的研究。我们从研究角度分析了WEDGE,验证了其在极端天气自动驾驶感知中的有效性。我们建立了分类和检测的基线性能,测试准确率为53.87%,mAP为45.41。最重要的是,WEDGE可用于微调最先进的检测器,在真实世界天气基准(如DAWN)上,对于卡车等生成良好的类别,将SOTA性能提升了4.48 AP。WEDGE是根据OpenAI的使用条款收集的,并在CC BY-NC-SA 4.0许可下公开发布。该工作和数据集的仓库可在https://infernolia.github.io/WEDGE获取。

英文摘要:

The open road poses many challenges to autonomous perception, including poor visibility from extreme weather conditions. Models trained on good-weather datasets frequently fail at detection in these out-of-distribution settings. To aid adversarial robustness in perception, we introduce WEDGE (WEather images by DALL-E GEneration): a synthetic dataset generated with a vision-language generative model via prompting. WEDGE consists of 3360 images in 16 extreme weather conditions manually annotated with 16513 bounding boxes, supporting research in the tasks of weather classification and 2D object detection. We have analyzed WEDGE from research standpoints, verifying its effectiveness for extreme-weather autonomous perception. We establish baseline performance for classification and detection with 53.87% test accuracy and 45.41 mAP. Most importantly, WEDGE can be used to fine-tune state-of-the-art detectors, improving SOTA performance on real-world weather benchmarks (such as DAWN) by 4.48 AP for well-generated classes like trucks. WEDGE has been collected under OpenAI's terms of use and is released for public use under the CC BY-NC-SA 4.0 license. The repository for this work and dataset is available at https://infernolia.github.io/WEDGE.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑