Skip to content

zengwb-lx/Face-Attributes-Mobile

 
 

Repository files navigation

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

image

Age

image

Euler angles:

image

Pitch:

image

Yaw:

image

Roll:

image

Beauty

image

Expression

image

Face shape

image

Face type

image

Glasses

image

Race

image

Dependencies

  • Python 3.6.8
  • PyTorch 1.3.0

Usage

Train

$ python train.py

To visualize the training process:

$ tensorboard --logdir=runs

image

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

About

Regress Face Attributes with MobileNetV2

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 100.0%