forked from wbenbihi/hourglasstensorflow
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconfig.cfg
32 lines (32 loc) · 817 Bytes
/
config.cfg
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
[DataSetHG]
training_txt_file: 'dataset.txt'
img_directory: 'F:/Cours/DHPE/mpii_human_pose_v1/images'
img_size: 256
hm_size: 64
num_joints: 16
remove_joints: None
joint_list = ['r_anckle', 'r_knee', 'r_hip', 'l_hip', 'l_knee', 'l_anckle', 'pelvis', 'thorax', 'neck', 'head', 'r_wrist', 'r_elbow', 'r_shoulder', 'l_shoulder', 'l_elbow', 'l_wrist']
[Network]
name: 'hg_refined_200'
nFeats: 256
nStacks: 4
nModules: 1
tiny: False
nLow: 4
dropout_rate: 0.2
mcam: False
[Train]
batch_size: 4
nEpochs: 200
epoch_size: 1000
learning_rate: 0.00025
learning_rate_decay: 0.96
decay_step: 2000
weighted_loss: False
[Validation]
valid_iteration: 10
[Saver]
log_dir_train: 'F:/Cours/DHPE/DHPE/hourglass_tiny/'
log_dir_test: 'F:/Cours/DHPE/DHPE/hourglass_tiny/'
saver_step: 500
saver_directory: ''