forked from foamliu/Face-Attributes-Mobile
-
Notifications
You must be signed in to change notification settings - Fork 0
/
README.t
92 lines (56 loc) · 3.85 KB
/
README.t
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
# 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](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/gender_dist.png)
### Age
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/age_dist.png)
### Euler angles:
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/euler_angles.png)
Pitch:
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/angle_pitch_dist.png)
Yaw:
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/angle_yaw_dist.png)
Roll:
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/angle_roll_dist.png)
### Beauty
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/beauty_dist.png)
### Expression
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/expression_dist.png)
### Face shape
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/face_shape_dist.png)
### Face type
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/face_type_dist.png)
### Glasses
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/glasses_dist.png)
### Race
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/race_dist.png)
## Dependencies
- Python 3.6.8
- PyTorch 1.3.0
## Usage
### Train
```bash
$ python train.py
```
To visualize the training process:
```bash
$ tensorboard --logdir=runs
```
![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/learning_curve.jpg)
Image | Aligned | Out | True |
|---|---|---|---|
|![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/0_raw.jpg)|![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/0_img.jpg)|$(result_out_0)|$(result_true_0)|
|![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/1_raw.jpg)|![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/1_img.jpg)|$(result_out_1)|$(result_true_1)|
|![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/2_raw.jpg)|![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/2_img.jpg)|$(result_out_2)|$(result_true_2)|
|![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/3_raw.jpg)|![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/3_img.jpg)|$(result_out_3)|$(result_true_3)|
|![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/4_raw.jpg)|![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/4_img.jpg)|$(result_out_4)|$(result_true_4)|
|![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/5_raw.jpg)|![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/5_img.jpg)|$(result_out_5)|$(result_true_5)|
|![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/6_raw.jpg)|![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/6_img.jpg)|$(result_out_6)|$(result_true_6)|
|![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/7_raw.jpg)|![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/7_img.jpg)|$(result_out_7)|$(result_true_7)|
|![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/8_raw.jpg)|![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/8_img.jpg)|$(result_out_8)|$(result_true_8)|
|![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/9_raw.jpg)|![image](https://github.com/foamliu/Face-Attributes-v2/raw/master/images/9_img.jpg)|$(result_out_9)|$(result_true_9)|