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
@by write xiaoshen 具体的解释位于:https://zhuanlan.zhihu.com/p/32866194?utm_source=qq&utm_medium=social 此外针对人脸的年龄检测可参考:https://data.vision.ee.ethz.ch/cvl/rrothe/imdb-wiki/