Regress Face Attributes with MobileNetV2.
- Estimate Gender, Age, Euler angles, Beauty, Expression, Face shape, Face type and Glasses with a single image.
- Lightweight: Params size (MB): 2.14, FLOPs size (GB): 0.32, Total Size (MB): 9.18.
CASIA WebFace DataSet, 479,653 faces.
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/gender_dist.png?raw=true)
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/age_dist.png?raw=true)
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/euler_angles.png?raw=true)
Pitch:
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/angle_pitch_dist.png?raw=true)
Yaw:
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/angle_yaw_dist.png?raw=true)
Roll:
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/angle_roll_dist.png?raw=true)
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/beauty_dist.png?raw=true)
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/expression_dist.png?raw=true)
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/face_shape_dist.png?raw=true)
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/face_type_dist.png?raw=true)
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/glasses_dist.png?raw=true)
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/race_dist.png?raw=true)
- Python 3.6.8
- PyTorch 1.3.0
To visualize the training process:
$ tensorboard --logdir=runs
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/learning_curve.jpg?raw=true)
Image |
Aligned |
Out |
True |
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/0_raw.jpg?raw=true) |
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/0_img.jpg?raw=true) |
age: 31 pitch: -5.58 roll: 0.83 yaw: -24.83 beauty: 47.2 expression: none gender: female glasses: none race: white |
age: 32 pitch: -4.77 roll: 1.15 yaw: -26.22 beauty: 57.96 expression: none gender: female glasses: none race: white |
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/1_raw.jpg?raw=true) |
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/1_img.jpg?raw=true) |
age: 32 pitch: 10.87 roll: -4.92 yaw: -20.07 beauty: 66.32 expression: none gender: male glasses: none race: white |
age: 31 pitch: 13.28 roll: -5.81 yaw: -18.85 beauty: 65.91 expression: none gender: male glasses: none race: white |
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/2_raw.jpg?raw=true) |
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/2_img.jpg?raw=true) |
age: 38 pitch: 12.82 roll: -0.2 yaw: -12.13 beauty: 35.49 expression: none gender: male glasses: none race: white |
age: 42 pitch: 13.01 roll: -0.03 yaw: -16.77 beauty: 50.19 expression: none gender: male glasses: none race: white |
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/3_raw.jpg?raw=true) |
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/3_img.jpg?raw=true) |
age: 22 pitch: 4.57 roll: -7.73 yaw: 23.17 beauty: 43.42 expression: none gender: female glasses: none race: white |
age: 23 pitch: 5.51 roll: -7.73 yaw: 18.73 beauty: 45.36 expression: none gender: female glasses: sun race: white |
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/4_raw.jpg?raw=true) |
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/4_img.jpg?raw=true) |
age: 34 pitch: 8.17 roll: -5.39 yaw: -0.97 beauty: 60.92 expression: none gender: male glasses: none race: white |
age: 37 pitch: 7.48 roll: -6.35 yaw: -0.35 beauty: 62.51 expression: none gender: male glasses: none race: white |
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/5_raw.jpg?raw=true) |
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/5_img.jpg?raw=true) |
age: 28 pitch: 9.68 roll: 11.88 yaw: -40.47 beauty: 49.96 expression: none gender: female glasses: none race: white |
age: 28 pitch: 7.41 roll: 12.21 yaw: -38.69 beauty: 48.13 expression: none gender: female glasses: none race: white |
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/6_raw.jpg?raw=true) |
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/6_img.jpg?raw=true) |
age: 35 pitch: 17.0 roll: -5.98 yaw: -3.54 beauty: 70.58 expression: none gender: female glasses: none race: white |
age: 37 pitch: 17.99 roll: -4.34 yaw: -7.96 beauty: 66.77 expression: none gender: female glasses: none race: white |
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/7_raw.jpg?raw=true) |
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/7_img.jpg?raw=true) |
age: 30 pitch: 20.61 roll: 12.72 yaw: -22.68 beauty: 59.12 expression: smile gender: male glasses: none race: white |
age: 28 pitch: 19.27 roll: 13.28 yaw: -26.46 beauty: 58.69 expression: smile gender: male glasses: none race: white |
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/8_raw.jpg?raw=true) |
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/8_img.jpg?raw=true) |
age: 46 pitch: 11.58 roll: -3.16 yaw: 19.05 beauty: 56.29 expression: none gender: male glasses: none race: white |
age: 45 pitch: 8.26 roll: -1.08 yaw: 20.05 beauty: 61.68 expression: none gender: male glasses: none race: white |
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/9_raw.jpg?raw=true) |
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/9_img.jpg?raw=true) |
age: 34 pitch: 17.02 roll: -1.11 yaw: -3.17 beauty: 62.15 expression: none gender: male glasses: none race: white |
age: 34 pitch: 18.41 roll: -0.76 yaw: -5.72 beauty: 55.57 expression: none gender: male glasses: none race: white |