Chinese Journal of Liquid Crystals and Displays, Volume. 37, Issue 6, 746(2022)

Cycle generative adversarial network guided by dual special attention mechanism

Jun-ming LAO1, Wu-jian YE2、*, Yi-jun LIU2, and Kai-yi YUAN1
Author Affiliations
  • 1College of Information Engineering, Guangdong University of Technology, Guangzhou 510006, China
  • 2College of Integrated Circuit,Guangdong University of Technology,Guangzhou 510006,China
  • show less

    The existing image generation methods based on cycle generative adversarial network have achieved excellent results in unpaired image to image translation tasks by introducing a separate generic attention module, but they also increase the model complexity and training time, and it is difficult to focus on all the details of key regions in the image, and there is still room for improvement in image generation results. To solve these problems, this paper proposes a dual special attention-mechanism guided cycle generative adversarial network architecture for unpaired image transformation translation (Dual-SAG-CycleGAN). Firstly, to improve the quality of the generated images and reduce the complexity of the network, a special attention module called SAG (Special Attention-mechanism Guided) is proposed to guide the generator. Then, to suppress the generation of extraneous noise by the generator, a discriminator based on special attention mechanism of CAM (Class Activation Mapping)is introduced. Finally, a cyclic consistency loss function for the background mask is introduced to guide the network to generate a more accurate mask map, which can better aid for image translation. Experiments demonstrate that compared with similar existing models, our proposed method can reduce the maximum number of parameters by 32.8%, speed of train 34.5% faster, and generate higher quality images with a minimum KID and FID of 1.13 and 57.54, respectively.

    Tools

    Get Citation

    Copy Citation Text

    Jun-ming LAO, Wu-jian YE, Yi-jun LIU, Kai-yi YUAN. Cycle generative adversarial network guided by dual special attention mechanism[J]. Chinese Journal of Liquid Crystals and Displays, 2022, 37(6): 746

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category:

    Received: Nov. 24, 2021

    Accepted: --

    Published Online: Jun. 22, 2022

    The Author Email: Wu-jian YE (yewjian@126.com)

    DOI:10.37188/CJLCD.2021-0293

    Topics