An image segmentation algorithm based on CycleGAN-vd is proposed to improve the poor robustness and image tearing of the CycleGAN algorithm in this paper. In CycleGAN-vd, the least squares loss function is used instead of the cross-entropy loss function for adversarial loss. In addition, the secondary adversarial loss is introduced into the generator network and the self-attention mechanism is integrated to weight the image features. Experimental results show that the SSIM score of CycleGAN-vd is 2.6% higher than that of StyleGAN. Moreover, the FID and SSIM rankings of CycleGAN-vd are multiplied with the aid of 10.1% and 11.4% in contrast with CycleGAN. Finally, YOLOv3 is used to verify the availability of the generated images.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Autonomous Driving Image Optimization Model Based on CycleGAN-vd

  • Yuqi Ouyang,
  • Junjie Yang,
  • Xuan Li,
  • Hong Mo

摘要

An image segmentation algorithm based on CycleGAN-vd is proposed to improve the poor robustness and image tearing of the CycleGAN algorithm in this paper. In CycleGAN-vd, the least squares loss function is used instead of the cross-entropy loss function for adversarial loss. In addition, the secondary adversarial loss is introduced into the generator network and the self-attention mechanism is integrated to weight the image features. Experimental results show that the SSIM score of CycleGAN-vd is 2.6% higher than that of StyleGAN. Moreover, the FID and SSIM rankings of CycleGAN-vd are multiplied with the aid of 10.1% and 11.4% in contrast with CycleGAN. Finally, YOLOv3 is used to verify the availability of the generated images.