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

SUSTechGAN:面向自动驾驶恶劣条件目标检测的图像生成

SUSTechGAN: Image Generation for Object Detection in Adverse Conditions of Autonomous Driving

Southern University of Science and Technology(南方科技大学);Research Institute of Trustworthy Autonomous Systems(可信自主系统研究院);University of Macau(澳门大学)

Gongjin Lan, Yang Peng, Qi Hao, Chengzhong Xu

AI总结:

提出SUSTechGAN框架,利用双注意力模块、多尺度生成器和新损失函数生成恶劣条件驾驶图像,显著提升YOLOv5在雨天和夜晚的目标检测性能。

摘要:

自动驾驶显著受益于数据驱动的深度神经网络。然而,自动驾驶中的数据通常符合长尾分布,其中恶劣条件下的关键驾驶数据难以收集。尽管生成对抗网络(GANs)已被应用于为自动驾驶增强数据,但在恶劣条件下生成驾驶图像仍然具有挑战性。在这项工作中,我们提出了一个新颖的框架 SUSTechGAN,它具有定制的双注意力模块、多尺度生成器和一个新颖的损失函数,用于生成驾驶图像,以改善自动驾驶在恶劣条件下的目标检测。我们测试了 SUSTechGAN 和知名的 GANs,以生成雨天和夜晚恶劣条件下的驾驶图像,并将生成的图像应用于重新训练目标检测网络。具体来说,我们将生成的图像添加到训练数据集中,以重新训练知名的 YOLOv5,并评估重新训练的 YOLOv5 在恶劣条件下进行目标检测的改进情况。实验结果表明,我们的 SUSTechGAN 生成的驾驶图像显著提高了重新训练的 YOLOv5 在雨天和夜晚条件下的性能,优于知名的 GANs。开源代码、视频描述和数据集可在第 1 页获取,以促进自动驾驶在恶劣条件下的图像生成发展。

英文摘要:

Autonomous driving significantly benefits from data-driven deep neural networks. However, the data in autonomous driving typically fits the long-tailed distribution, in which the critical driving data in adverse conditions is hard to collect. Although generative adversarial networks (GANs) have been applied to augment data for autonomous driving, generating driving images in adverse conditions is still challenging. In this work, we propose a novel framework, SUSTechGAN, with customized dual attention modules, multi-scale generators, and a novel loss function to generate driving images for improving object detection of autonomous driving in adverse conditions. We test the SUSTechGAN and the well-known GANs to generate driving images in adverse conditions of rain and night and apply the generated images to retrain object detection networks. Specifically, we add generated images into the training datasets to retrain the well-known YOLOv5 and evaluate the improvement of the retrained YOLOv5 for object detection in adverse conditions. The experimental results show that the generated driving images by our SUSTechGAN significantly improved the performance of retrained YOLOv5 in rain and night conditions, which outperforms the well-known GANs. The open-source code, video description and datasets are available on the page 1 to facilitate image generation development in autonomous driving under adverse conditions.

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

↑