Sawyer260/Face-Attributes-Mobile

Regress Face Attributes with MobileNetV2

Face Attributes Mobile

Regress Face Attributes with MobileNetV2.

Features

1. Estimate Gender, Age, Euler angles, Beauty, Expression, Face shape, Face type and Glasses with a single image.
2. Lightweight: Params size (MB): 2.14, FLOPs size (GB): 0.32, Total Size (MB): 9.18.

DataSet

CASIA WebFace DataSet, 479,653 faces.

Gender

Age

Euler angles:

Pitch:

Yaw:

Roll:

Beauty

Expression

Face shape

Face type

Glasses

Race

Dependencies

• Python 3.6.8
• PyTorch 1.3.0

Usage

Train

$ python train.py

To visualize the training process：

$ tensorboard --logdir=runs

Image	Aligned	Out	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
		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
		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
		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
		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
		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
		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
		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
		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
		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