首页 | 本学科首页   官方微博 | 高级检索  
     

基于改进CycleGAN的道路场景语义分割研究
引用本文:张如涛,黄山,汪鸿浩.基于改进CycleGAN的道路场景语义分割研究[J].计算机工程与应用,2022,58(15):278-284.
作者姓名:张如涛  黄山  汪鸿浩
作者单位:四川大学 电气工程学院,成都 610065
摘    要:道路场景下的语义分割是无人驾驶中关键的技术,也是计算机视觉中重要的一个领域,而传统的语义分割方法需要对训练数据进行像素级的标注,对数据的要求极高。针对这一问题,将改进的循环生成对抗网络(cycle-consistent adversarial networks,CycleGAN)用于道路场景语义分割,该网络避免了大量的像素级标注且不需要成对的数据集,降低了数据集的要求。将原网络的目标函数用最小二乘损失和Smooth L1范数替代,增加了网络训练的稳定性且提高了生成图像的质量,并引入特征损失保证图像特征的保留,使得生成图像更加真实。使用道路场景分割中常用的Cityscapes数据集进行实验,并用语义分割领域常用的性能评价指标验证了方法的有效性,实验结果表明相较于原网络各性能都有一定提升。

关 键 词:语义分割  循环生成对抗网络  损失函数  图像生成  

Research on Road Scene Semantic Segmentation Based on Improved CycleGAN
ZHANG Rutao,HUANG Shan,WANG Honghao.Research on Road Scene Semantic Segmentation Based on Improved CycleGAN[J].Computer Engineering and Applications,2022,58(15):278-284.
Authors:ZHANG Rutao  HUANG Shan  WANG Honghao
Affiliation:College of Electrical Engineering, Sichuan University, Chengdu 610065, China
Abstract:Semantic segmentation in road scenes is a key technology in unmanned driving, and it is also an important field in computer vision, but traditional semantic segmentation methods require pixel-level labeling of training data, which has extremely high data requirements. To address this problem, an improved cycle-consistent adversarial network is used for semantic segmentation, the network avoids a large number of pixel-level annotations and does not require paired data sets, which reduces the requirements of data sets. Replace the objective function of the original network with least square loss and Smooth L1 norm, which increases the stability of network training and improves the quality of the generated image, the introduction of identity loss ensures the preservation of image features, making the generated image more realistic. The cityscapes dataset commonly used in road segmentation is used to conduct experiments, and the effectiveness of the method is verified by performance evaluation indicators commonly used in segmentation, the experimental results show that each performance has a certain improvement compared with the original network.
Keywords:semantic segmentation  cycle-consistent adversarial networks  loss function  image generate  
点击此处可从《计算机工程与应用》浏览原始摘要信息
点击此处可从《计算机工程与应用》下载全文
设为首页 | 免责声明 | 关于勤云 | 加入收藏

Copyright©北京勤云科技发展有限公司  京ICP备09084417号