Skip to content

The code of ICPR2020 paper "Attributes Aware Face Generation with Generative Adversarial Networks"

Notifications You must be signed in to change notification settings

huitailangyz/AFGAN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Attributes Aware Face Generation with Generative Adversarial Networks

This paper has been accepted by ICPR2020.

This repository is an official implementation of the paper Attributes Aware Face Generation with Generative Adversarial Networks.

Model

AFGAN

Abstract

Recent studies have shown remarkable success in face image generations. However, most of the existing methods only generate face images from random noise, and cannot generate face images according to the specific attributes. In this paper, we focus on the problem of face synthesis from attributes, which aims at generating faces with specific characteristics corresponding to the given attributes. To this end, we propose a novel attributes aware face image generator method with generative adversarial networks called AFGAN. Specifically, we firstly propose a two-path embedding layer and self-attention mechanism to convert binary attribute vector to rich attribute features. Then three stacked generators generate $64 \times 64$, $128 \times 128$ and $256 \times 256$ resolution face images respectively by taking the attribute features as input. In addition, an image-attribute matching loss is proposed to enhance the correlation between the generated images and input attributes. Extensive experiments on CelebA demonstrate the superiority of our AFGAN in terms of both qualitative and quantitative evaluations.

Usage

Train

cfg/CelebA_AFGAN.yml
    TRAIN.FLAG: Frue

python main.py --cfg=cfg/CelebA_AFGAN.yml --gpu=0 --data_dir=XXX

Test

cfg/CelebA_AFGAN.yml
    TRAIN.FLAG: False

python main.py --cfg=cfg/CelebA_AFGAN.yml --gpu=0 --data_dir=XXX

Model

We have uploaded the model of attribute classifier (AlexNet_epoch_99.pth) to google drive.

About

The code of ICPR2020 paper "Attributes Aware Face Generation with Generative Adversarial Networks"

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages