-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathfeature_celeba_organize.py
49 lines (43 loc) · 1.59 KB
/
feature_celeba_organize.py
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
""" module to get the desierble order of features """
import numpy as np
feature_name_celeba_org = [
'5_o_Clock_Shadow', 'Arched_Eyebrows', 'Attractive',
'Bags_Under_Eyes', 'Bald', 'Bangs', 'Big_Lips', 'Big_Nose',
'Black_Hair', 'Blond_Hair', 'Blurry', 'Brown_Hair',
'Bushy_Eyebrows', 'Chubby', 'Double_Chin', 'Eyeglasses', 'Goatee',
'Gray_Hair', 'Heavy_Makeup', 'High_Cheekbones', 'Male',
'Mouth_Slightly_Open', 'Mustache', 'Narrow_Eyes', 'No_Beard',
'Oval_Face', 'Pale_Skin', 'Pointy_Nose', 'Receding_Hairline',
'Rosy_Cheeks', 'Sideburns', 'Smiling', 'Straight_Hair',
'Wavy_Hair', 'Wearing_Earrings', 'Wearing_Hat', 'Wearing_Lipstick',
'Wearing_Necklace', 'Wearing_Necktie', 'Young'
]
feature_name_celeba_rename = [
'Shadow', 'Arched_Eyebrows', 'Attractive', 'Eye_bags', 'Bald',
'Bangs', 'Big_Lips', 'Big_Nose', 'Black_Hair', 'Blond_Hair',
'Blurry', 'Brown_Hair', 'Bushy_Eyebrows', 'Chubby', 'Double_Chin',
'Eyeglasses', 'Goatee', 'Gray_Hair', 'Makeup', 'High_Cheekbones',
'Male', 'Mouth_Open', 'Mustache', 'Narrow_Eyes', 'Beard',
'Oval_Face', 'Skin_Tone', 'Pointy_Nose', 'Hairline', 'Rosy_Cheeks',
'Sideburns', 'Smiling', 'Straight_Hair', 'Wavy_Hair', 'Earrings',
'Hat', 'Lipstick', 'Necklace', 'Necktie', 'Age'
]
feature_reverse = np.array([
1, 1, 1, 1, 1,
1, 1, 1, 1, 1,
1, 1, 1, 1, 1,
1, 1, 1, 1, 1,
1, 1, 1, 1,-1,
1,-1, 1, 1, 1,
1, 1, 1, 1, 1,
1, 1, 1, 1,-1
])
feature_celeba_layout = [
[20, 39, 26, ],
[5, 28, 4, ],
[7, 27, 18],
[31, 21, 33],
[24, 16, 30],
[9, 8, 17],
[15, 34, 38],
]