-
Notifications
You must be signed in to change notification settings - Fork 4
/
config.py
44 lines (29 loc) · 1006 Bytes
/
config.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
class DatasetName:
w300 = '300W'
cofw = 'COFW'
class ModelArch:
MNV2 = 'mobileNetV2'
EFNB0 = 'EfficientNet-B0'
EFNB3 = 'EfficientNet-B3'
class DatasetType:
data_type_train = 0
data_type_test = 1
class LearningConfig:
batch_size = 3
epochs = 150
class InputDataSize:
image_input_size = 224
class W300Conf:
W300W_prefix_path = './300W/'
train_annotation = W300W_prefix_path + 'train_set/annotations/'
train_image = W300W_prefix_path + 'train_set/images/'
test_annotation_path = W300W_prefix_path + 'test_set/annotations/'
test_image_path = W300W_prefix_path + 'test_set/images/'
num_of_landmarks = 68
class CofwConf:
cofw_prefix_path = './cofw/'
train_annotation = cofw_prefix_path + 'train_set/annotations/'
train_image = cofw_prefix_path + 'train_set/images/'
test_annotation_path = cofw_prefix_path + 'test_set/annotations/'
test_image_path = cofw_prefix_path + 'test_set/images/'
num_of_landmarks = 29