Facial attribute classification based on graph attention (tensorflow)
1.Datasets
CelebA(aligned & cropped version):http://mmlab.ie.cuhk.edu.hk/projects/CelebA.html
LFWA:http://vis-www.cs.umass.edu/lfw/
2.Pretrain Model
Alexnet with bn:converted from caffe model trained by Marcel Simon, Erik Rodner, Joachim Denzler.
You can download the caffe model here:
https://github.com/cvjena/cnn-models
We also provide the converted tensorflow model in the following link, along with the converted resnet50 mdoel trained on VggFace2.
链接:https://pan.baidu.com/s/1KBSN0ZGuGAtXRRyP7fq_7Q 提取码:xhzu
3.Data Augmentation
As the LFWA dataset is too small, we add some distortion to the training set. The distortion tool is from https://github.com/mdbloice/Augmentor
The paper has been uploaded on https://arxiv.org/abs/1810.09162.