-
Notifications
You must be signed in to change notification settings - Fork 145
/
256x192_adam_lr1e-3-hrw48_cam_2x_wo_pw3d.yaml
79 lines (79 loc) · 1.43 KB
/
256x192_adam_lr1e-3-hrw48_cam_2x_wo_pw3d.yaml
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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
DATASET:
DATASET: 'mix_smpl_cam'
SET_LIST:
- ROOT: './data/h36m/'
TEST_SET: 'Sample_20_test_Human36M_smpl'
TRAIN_SET: 'Sample_trainmin_train_Human36M_smpl_leaf_twist'
- ROOT: './data/coco/'
TRAIN_SET: 'train2017'
- ROOT: './data/3dhp/'
TRAIN_SET: 'train_v2'
PROTOCOL: 2
FLIP: True
ROT_FACTOR: 30
SCALE_FACTOR: 0.3
NUM_JOINTS_HALF_BODY: 8
PROB_HALF_BODY: -1
COLOR_FACTOR: 0.2
OCCLUSION: True
MODEL:
TYPE: 'HRNetSMPLCam'
HR_PRETRAINED: './pose_hrnet_w48_256x192.pth'
PRETRAINED: ''
# TRY_LOAD: ''
TRY_LOAD: ''
RESUME: ''
FOCAL_LENGTH: 1000
IMAGE_SIZE:
- 256
- 256
HEATMAP_SIZE:
- 64
- 64
NUM_JOINTS: 29
HRNET_TYPE: 48
EXTRA:
SIGMA: 2
BACKBONE: 'resnet'
CROP: 'padding'
AUGMENT: 'none'
PRESET: 'simple_smpl_3d_cam'
DEPTH_DIM: 64
POST:
NORM_TYPE: 'softmax'
BBOX_3D_SHAPE:
- 2200
- 2200
- 2200
LOSS:
TYPE: 'LaplaceLossDimSMPLCam'
ELEMENTS:
BETA_WEIGHT: 1
BETA_REG_WEIGHT: 0
PHI_REG_WEIGHT: 0.0001
LEAF_REG_WEIGHT: 0
TWIST_WEIGHT: 1
THETA_WEIGHT: 0.01
UVD24_WEIGHT: 1
XYZ24_WEIGHT: 0
XYZ_SMPL24_WEIGHT: 0
XYZ_SMPL17_WEIGHT: 0
VERTICE_WEIGHT: 0
USE_LAPLACE: True
TEST:
HEATMAP2COORD: 'coord'
TRAIN:
WORLD_SIZE: 4
BATCH_SIZE: 36
BEGIN_EPOCH: 0
END_EPOCH: 200
OPTIMIZER: 'adam'
LR: 0.001
LR_FACTOR: 0.1
LR_STEP:
- 60
- 80
DPG_MILESTONE: 140
DPG_STEP:
- 160
- 190