节点文献

基于改进GAN的端到端自动驾驶图像生成方法

An Image Generation Method for Automated Driving Based on Improved GAN

  • 推荐 CAJ下载
  • PDF下载
  • 不支持迅雷等下载工具,请取消加速工具后下载。

【作者】 孙雄风黄珍陈志军罗鹏

【Author】 SUN Xiongfeng;HUANG Zhen;CHEN Zhijun;LUO Peng;School of Automation,Wuhan University of Technology;Intelligent Transportation Systems Research Center,Wuhan University of Technology;

【通讯作者】 黄珍;

【机构】 武汉理工大学自动化学院武汉理工大学智能交通系统研究中心

【摘要】 基于端到端数据系统的自动驾驶系统对驾驶图像存在巨大需求。为解决一般生成式对抗网络模型在扩充驾驶图像数据集时不稳定及生成图像特征缺乏多样性的问题,研究1种改进网络模型LS-InfoGAN。结合最小二乘对抗损失防止模型梯度消失,并缓解生成器优化矛盾,提升模型训练稳定性。通过最大化生成图像与真实图像间的互信息提升生成器特征学习能力,改善生成图像特征多样性。利用转置卷积层还原图像特征,提升生成图像特征清晰度。以自主构建的模拟驾驶场景中获取的带标签驾驶图像集对模型有效性及其数据集扩充应用效果进行验证。实验分析表明:相比改进前模型,LS-InfoGAN模型的图像生成过程稳定性平均提升35%;使用此模型扩充的数据集进行端到端自动驾驶系统中决策网络的训练能在不采集新图像的情况下将系统决策性能提升1%~2%;建议使用此模型扩充图像数据集时将生成图像数量设置为原始训练集图像数量的1~2倍。

【Abstract】 There is a huge demand for driving images in the automated driving systems based on end-to-end data sys-tem. In order to solve the instability of general generative adversarial network model and the lack of diversity of generat-ed image features when expanding the driving image data set,this work proposed an improved network model,LS-Info-GAN. The least-squares loss is used to prevent the model gradient from disappearing and alleviate the contradiction inthe generator during optimization,thereby improving stability of the model. The learning ability of the generator is im-proved by maximizing mutual information between generated images and actual images,thus improving the diversity ofits features. The transposed convolutional layer to restore the image features is used to improve the clarity of the generat-ed image features. The effectiveness and application performance of the model are verified with a labeled image datasetacquired in self-built driving scenes. According to the academic analysis in this study,compared with the model beforethe improvement,the stability of the image generation process of the LS-Info GAN model is improved by an average of 35%.Besides,when used for training in the decision network of end-to-end self-driving systems,the augmented data-set can improve the decision performance by 1% to 2% without acquiring new images. The recommended number ofgenerated images is 1 to 2 times the number of original images when the model is used to augment the dataset.

【基金】 国家自然科学基金项目(U1764262);湖北省自然科学基金项目(2017CFA008)资助
  • 【文献出处】 交通信息与安全 ,Journal of Transport Information and Safety , 编辑部邮箱 ,2021年05期
  • 【分类号】U463.6;TP391.41;TP183
  • 【下载频次】413
节点文献中: 

本文链接的文献网络图示:

本文的引文网络