File size: 1,006 Bytes
ad1e0a0 |
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 |