arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2408.07239cs.CVcs.AIcs.LG
收藏

在语义分割训练中通过图像增强提升恶劣天气下的自动驾驶车辆感知能力

Enhancing Autonomous Vehicle Perception in Adverse Weather through Image Augmentation during Semantic Segmentation Training

Henry M. Gunn High School(亨利·M·冈恩高中);UC Berkeley(加州大学伯克利分校);University of Michigan(密歇根大学)

Ethan Kou, Noah Curran

AI总结:

针对仅晴天数据训练的语义分割模型恶劣天气性能差的问题,本研究用CARLA采集数据,在UNet训练中加入雨、雾、亮度增强,证实其可提升夜间恶劣天气分割性能,但仍不及真实天气数据训练的模型,为后续域适应优化提供方向。

摘要:

鲁棒感知对自动驾驶车辆的导航与定位至关重要。语义分割等视觉处理任务需在不同天气条件及不同时段下正常工作。语义分割指为每个像素分配一个类别,可用于定位整体特征(1)。训练分割模型需要大量数据,而分割数据的标注过程尤为繁琐。此外,许多大型数据集仅包含晴朗天气拍摄的图像。这会带来一个问题:仅用晴朗天气数据训练的模型,在雾天或雨天等恶劣天气条件下的性能会受到影响。我们提出假设:若仅有晴天图像数据集,在训练过程中应用图像增强(如随机降雨、起雾和亮度调整)可实现向多样天气条件的域适应。我们使用CARLA(一款3D写实自动驾驶车辆模拟器),从10个不同城镇采集了包含29个类别的1200张晴朗天气图像(2),还采集了1200张带有随机天气效果的图像。我们训练了编码器-解码器结构的UNet模型来执行语义分割任务。应用增强方法显著提升了模型在夜间恶劣天气条件下的分割性能(p < 0.001)。然而,除晴天外,在所有条件下,用真实天气数据训练的模型损失均显著低于用增强数据训练的模型。这表明该域适应方法仍有提升空间。未来工作应测试更多类型的增强方式,并使用真实图像替代CARLA生成的图像。理想情况下,增强训练的模型性能应达到或超过天气数据训练模型的性能。

英文摘要:

Robust perception is crucial in autonomous vehicle navigation and localization. Visual processing tasks, like semantic segmentation, should work in varying weather conditions and during different times of day. Semantic segmentation is where each pixel is assigned a class, which is useful for locating overall features (1). Training a segmentation model requires large amounts of data, and the labeling process for segmentation data is especially tedious. Additionally, many large datasets include only images taken in clear weather. This is a problem because training a model exclusively on clear weather data hinders performance in adverse weather conditions like fog or rain. We hypothesize that given a dataset of only clear days images, applying image augmentation (such as random rain, fog, and brightness) during training allows for domain adaptation to diverse weather conditions. We used CARLA, a 3D realistic autonomous vehicle simulator, to collect 1200 images in clear weather composed of 29 classes from 10 different towns (2). We also collected 1200 images of random weather effects. We trained encoder-decoder UNet models to perform semantic segmentation. Applying augmentations significantly improved segmentation under weathered night conditions (p < 0.001). However, models trained on weather data have significantly lower losses than those trained on augmented data in all conditions except for clear days. This shows there is room for improvement in the domain adaptation approach. Future work should test more types of augmentations and also use real-life images instead of CARLA. Ideally, the augmented model meets or exceeds the performance of the weather model.

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

↑