config.cfg

[DataSetHG]
training_data_file: 'train/Annotations/train.csv'
testing_data_file: 'test/test.csv'
load_file: 'hg_refined'
train_img_directory: 'train'
test_img_directory: 'test'
img_size: 256
hm_size: 64
dress_type: 'outwear'
[blouse]
num_joints: 13
model_file: 'saver/blouse/'
joint_list = ['neckline_left', 'neckline_right', 'shoulder_left', 'shoulder_right', 'center_front', 'armpit_left', 'armpit_right', 'top_hem_left', 'top_hem_right', 'cuff_left_in', 'cuff_left_out', 'cuff_right_in', 'cuff_right_out']
[outwear]
num_joints: 14
model_file: 'saver/outwear/'
joint_list = ['neckline_left','neckline_right','shoulder_left','shoulder_right','armpit_left','armpit_right','waistline_left','waistline_right','cuff_left_in','cuff_left_out','cuff_right_in','cuff_right_out','top_hem_left','top_hem_right']
[trousers]
num_joints: 7
model_file: 'saver/trousers/'
joint_list = ['waistband_left','waistband_right','crotch','bottom_left_in','bottom_left_out','bottom_right_in','bottom_right_out']
[skirt]
num_joints: 4
model_file: 'saver/skirt/'
joint_list = ['waistband_left','waistband_right','hemline_left','hemline_right']
[dress]
num_joints: 15
model_file: 'saver/dress/'
joint_list = ['neckline_left','neckline_right','shoulder_left','shoulder_right','center_front','armpit_left','armpit_right','waistline_left','waistline_right','cuff_left_in','cuff_left_out','cuff_right_in','cuff_right_out','hemline_left','hemline_right']
[Network]
name: 'blouse/hg_refined_200'
nFeats: 256
nStacks: 4
nModules: 1
tiny: False
nLow: 4
dropout_rate: 0.2
mcam: False
[Train]
batch_size: 16
nEpochs: 30
epoch_size: 1000
learning_rate: 0.00025
learning_rate_decay: 0.96
decay_step: 2000
weighted_loss: True
[Validation]
valid_iteration: 10
[Saver]
log_dir_train: 'train_summary'
log_dir_test: 'test_summary'
saver_step: 500
saver_directory: 'saver'