EdgeGAN: One-way mapping generative adversarial network based on the edge information for unpaired training set

Yijie Li, Qiaokang Liang, Zhengwei Li, Youcheng Lei, Wei Sun, Yaonan Wang, Dan Zhang

Research output: Journal article publicationJournal articleAcademic researchpeer-review

1 Citation (Scopus)

Abstract

Image conversion has attracted mounting attention due to its practical applications. This paper proposes a lightweight network structure that can implement unpaired training sets to complete one-way image mapping, based on the generative adversarial network (GAN) and a fixed-parameter edge detection convolution kernel. Compared with the cycle consistent adversarial network (CycleGAN), the proposed network features simpler structure, fewer parameters (only 37.48% of the parameters in CycleGAN), and less training cost (only 35.47% of the GPU memory usage and 17.67% of the single iteration time in CycleGAN). Remarkably, the cyclic consistency becomes not mandatory for ensuring the consistency of the content before and after image mapping. This network has achieved significant processing effects in some image translation tasks, and its effectiveness and validity have been well demonstrated through typical experiments. In the quantitative classification evaluation based on VGG-16, the algorithm proposed in this paper has achieved superior performance.

Original languageEnglish
Article number103187
JournalJournal of Visual Communication and Image Representation
Volume78
DOIs
Publication statusPublished - Jul 2021
Externally publishedYes

Keywords

  • Image conversion
  • Image-to-image translation
  • Lightweight generative adversarial network
  • Unpaired image-to-image translation

ASJC Scopus subject areas

  • Signal Processing
  • Media Technology
  • Computer Vision and Pattern Recognition
  • Electrical and Electronic Engineering

Fingerprint

Dive into the research topics of 'EdgeGAN: One-way mapping generative adversarial network based on the edge information for unpaired training set'. Together they form a unique fingerprint.

Cite this