|
class DatasetName: |
|
w300 = 'w300' |
|
cofw = 'cofw' |
|
wflw = 'wflw' |
|
|
|
class DatasetType: |
|
data_type_train = 0 |
|
data_type_validation = 1 |
|
data_type_test = 2 |
|
|
|
class LearningConfig: |
|
batch_size = 70 |
|
epochs = 150 |
|
|
|
class InputDataSize: |
|
image_input_size = 224 |
|
|
|
class WflwConf: |
|
Wflw_prefix_path = './data/wflw/' |
|
|
|
''' augmented version''' |
|
augmented_train_pose = Wflw_prefix_path + 'training_set/augmented/pose/' |
|
augmented_train_annotation = Wflw_prefix_path + 'training_set/augmented/annotations/' |
|
augmented_train_atr = Wflw_prefix_path + 'training_set/augmented/atrs/' |
|
augmented_train_image = Wflw_prefix_path + 'training_set/augmented/images/' |
|
augmented_train_tf_path = Wflw_prefix_path + 'training_set/augmented/tf/' |
|
''' original version''' |
|
no_aug_train_annotation = Wflw_prefix_path + 'training_set/no_aug/annotations/' |
|
no_aug_train_atr = Wflw_prefix_path + 'training_set/no_aug/atrs/' |
|
no_aug_train_pose = Wflw_prefix_path + 'training_set/no_aug/pose/' |
|
no_aug_train_image = Wflw_prefix_path + 'training_set/no_aug/images/' |
|
no_aug_train_tf_path = Wflw_prefix_path + 'training_set/no_aug/tf/' |
|
|
|
orig_number_of_training = 7500 |
|
orig_number_of_test = 2500 |
|
|
|
augmentation_factor = 4 |
|
augmentation_factor_rotate = 15 |
|
num_of_landmarks = 98 |
|
|
|
class CofwConf: |
|
Cofw_prefix_path = './data/cofw/' |
|
|
|
augmented_train_pose = Cofw_prefix_path + 'training_set/augmented/pose/' |
|
augmented_train_annotation = Cofw_prefix_path + 'training_set/augmented/annotations/' |
|
augmented_train_image = Cofw_prefix_path + 'training_set/augmented/images/' |
|
augmented_train_tf_path = Cofw_prefix_path + 'training_set/augmented/tf/' |
|
|
|
orig_number_of_training = 1345 |
|
orig_number_of_test = 507 |
|
|
|
augmentation_factor = 10 |
|
num_of_landmarks = 29 |
|
|
|
|
|
class D300wConf: |
|
w300w_prefix_path = './data/300W/' |
|
|
|
orig_300W_train = w300w_prefix_path + 'orig_300W_train/' |
|
augmented_train_pose = w300w_prefix_path + 'training_set/augmented/pose/' |
|
augmented_train_annotation = w300w_prefix_path + 'training_set/augmented/annotations/' |
|
augmented_train_image = w300w_prefix_path + 'training_set/augmented/images/' |
|
augmented_train_tf_path = w300w_prefix_path + 'training_set/augmented/tf/' |
|
|
|
no_aug_train_annotation = w300w_prefix_path + 'training_set/no_aug/annotations/' |
|
no_aug_train_pose = w300w_prefix_path + 'training_set/no_aug/pose/' |
|
no_aug_train_image = w300w_prefix_path + 'training_set/no_aug/images/' |
|
no_aug_train_tf_path = w300w_prefix_path + 'training_set/no_aug/tf/' |
|
|
|
orig_number_of_training = 3148 |
|
orig_number_of_test_full = 689 |
|
orig_number_of_test_common = 554 |
|
orig_number_of_test_challenging = 135 |
|
|
|
augmentation_factor = 4 |
|
num_of_landmarks = 68 |
|
|
|
|