Hugging Face's logo

AlexZheng
/
alphapose

Model card Files
xet
 Community
2
AlexZheng commited on
Commit
97ed04d
·
1 Parent(s): a63ba64

Upload 2 files

Browse files
Files changed (2) hide show
  1. 256x192_res152_lr1e-3_1x-duc.yaml +73 -0
  2. 256x192_res50_lr1e-3_2x-dcn.yaml +74 -0
256x192_res152_lr1e-3_1x-duc.yaml ADDED
@@ -0,0 +1,73 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ DATASET:
2
+ TRAIN:
3
+ TYPE: 'Mscoco'
4
+ ROOT: './data/coco/'
5
+ IMG_PREFIX: 'train2017'
6
+ ANN: 'annotations/person_keypoints_train2017.json'
7
+ AUG:
8
+ FLIP: true
9
+ ROT_FACTOR: 40
10
+ SCALE_FACTOR: 0.3
11
+ NUM_JOINTS_HALF_BODY: 8
12
+ PROB_HALF_BODY: -1
13
+ VAL:
14
+ TYPE: 'Mscoco'
15
+ ROOT: './data/coco/'
16
+ IMG_PREFIX: 'val2017'
17
+ ANN: 'annotations/person_keypoints_val2017.json'
18
+ TEST:
19
+ TYPE: 'Mscoco_det'
20
+ ROOT: './data/coco/'
21
+ IMG_PREFIX: 'val2017'
22
+ DET_FILE: './exp/json/test_det_yolo.json'
23
+ ANN: 'annotations/person_keypoints_val2017.json'
24
+ DATA_PRESET:
25
+ TYPE: 'simple'
26
+ SIGMA: 2
27
+ NUM_JOINTS: 17
28
+ IMAGE_SIZE:
29
+ - 256
30
+ - 192
31
+ HEATMAP_SIZE:
32
+ - 64
33
+ - 48
34
+ MODEL:
35
+ TYPE: 'FastPose_DUC'
36
+ BACKBONE: 'se-resnet'
37
+ PRETRAINED: ''
38
+ TRY_LOAD: ''
39
+ NUM_DECONV_FILTERS:
40
+ - 256
41
+ - 256
42
+ - 256
43
+ NUM_LAYERS: 152
44
+ FINAL_CONV_KERNEL: 1
45
+ STAGE1:
46
+ NUM_CONV: 4
47
+ STAGE2:
48
+ NUM_CONV: 2
49
+ STAGE3:
50
+ NUM_CONV: 1
51
+ LOSS:
52
+ TYPE: 'MSELoss'
53
+ DETECTOR:
54
+ NAME: 'yolo'
55
+ CONFIG: 'detector/yolo/cfg/yolov3-spp.cfg'
56
+ WEIGHTS: 'detector/yolo/data/yolov3-spp.weights'
57
+ NMS_THRES: 0.6
58
+ CONFIDENCE: 0.05
59
+ TRAIN:
60
+ WORLD_SIZE: 4
61
+ BATCH_SIZE: 32
62
+ BEGIN_EPOCH: 0
63
+ END_EPOCH: 200
64
+ OPTIMIZER: 'adam'
65
+ LR: 0.001
66
+ LR_FACTOR: 0.1
67
+ LR_STEP:
68
+ - 90
69
+ - 120
70
+ DPG_MILESTONE: 140
71
+ DPG_STEP:
72
+ - 160
73
+ - 190
256x192_res50_lr1e-3_2x-dcn.yaml ADDED
@@ -0,0 +1,74 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ DATASET:
2
+ TRAIN:
3
+ TYPE: 'Mscoco'
4
+ ROOT: './data/coco/'
5
+ IMG_PREFIX: 'train2017'
6
+ ANN: 'annotations/person_keypoints_train2017.json'
7
+ AUG:
8
+ FLIP: true
9
+ ROT_FACTOR: 45
10
+ SCALE_FACTOR: 0.35
11
+ NUM_JOINTS_HALF_BODY: 8
12
+ PROB_HALF_BODY: 0.3
13
+ VAL:
14
+ TYPE: 'Mscoco'
15
+ ROOT: './data/coco/'
16
+ IMG_PREFIX: 'val2017'
17
+ ANN: 'annotations/person_keypoints_val2017.json'
18
+ TEST:
19
+ TYPE: 'Mscoco_det'
20
+ ROOT: './data/coco/'
21
+ IMG_PREFIX: 'val2017'
22
+ DET_FILE: './exp/json/test_det_yolo.json'
23
+ ANN: 'annotations/person_keypoints_val2017.json'
24
+ DATA_PRESET:
25
+ TYPE: 'simple'
26
+ SIGMA: 2
27
+ NUM_JOINTS: 17
28
+ IMAGE_SIZE:
29
+ - 256
30
+ - 192
31
+ HEATMAP_SIZE:
32
+ - 64
33
+ - 48
34
+ MODEL:
35
+ TYPE: 'FastPose'
36
+ PRETRAINED: ''
37
+ TRY_LOAD: ''
38
+ NUM_DECONV_FILTERS:
39
+ - 256
40
+ - 256
41
+ - 256
42
+ NUM_LAYERS: 50
43
+ DCN:
44
+ MODULATED: false
45
+ DEFORM_GROUP: 1
46
+ FALLBACK_ON_STRIDE: false
47
+ STAGE_WITH_DCN:
48
+ - false
49
+ - true
50
+ - true
51
+ - true
52
+ LOSS:
53
+ TYPE: 'MSELoss'
54
+ DETECTOR:
55
+ NAME: 'yolo'
56
+ CONFIG: 'detector/yolo/cfg/yolov3-spp.cfg'
57
+ WEIGHTS: 'detector/yolo/data/yolov3-spp.weights'
58
+ NMS_THRES: 0.6
59
+ CONFIDENCE: 0.05
60
+ TRAIN:
61
+ WORLD_SIZE: 4
62
+ BATCH_SIZE: 32
63
+ BEGIN_EPOCH: 0
64
+ END_EPOCH: 270
65
+ OPTIMIZER: 'adam'
66
+ LR: 0.001
67
+ LR_FACTOR: 0.1
68
+ LR_STEP:
69
+ - 170
70
+ - 200
71
+ DPG_MILESTONE: 210
72
+ DPG_STEP:
73
+ - 230
74
+ - 250