Six face attributes (Attractive, EyeGlasses, Male, MouthOpen, Smiling, Young) predication from a single face image.
PyTorch implementation. Trained using celebA dataset.
- Python 3.6+ (Anaconda)
- PyTorch-0.2.0 +
- scipy, numpy, sklearn etc.
- OpenCV3 (Python)
Tested on Ubuntu 14.04 LTS, Python 3.6 (Anaconda), PyTorch-0.3.0, CUDA 8.0, cuDNN 5.0
detMTCNN_celebA.py
AttrListGen.py
dataloadercelebA.py
TrainAttrPreRes18V0.py
TrainAttrPreV0.py
AttrPreModelRes18_256V0.py
AttrPreModelRes34_256V0.py
AttrEvaRes18_256V0.py
AttrEvaRes34_256V0.py
Focal Loss:
Attractive 0.8231
EyeGlasses 0.9980
Male 0.9721
MouthOpen 0.9407
Smiling 0.9200
Young 0.8776