Skip to content

Commit f8897aa

Browse files
author
Bin Xiao
committed
add config files for coco resnet
1 parent 1cc885b commit f8897aa

6 files changed

+498
-0
lines changed

‎experiments/coco/resnet/res101_256x192_d256x3_adam_lr1e-3.yaml

+83
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,83 @@
1+
AUTO_RESUME: true
2+
CUDNN:
3+
BENCHMARK: true
4+
DETERMINISTIC: false
5+
ENABLED: true
6+
DATA_DIR: ''
7+
GPUS: (0,1,2,3)
8+
OUTPUT_DIR: 'output'
9+
LOG_DIR: 'log'
10+
WORKERS: 24
11+
PRINT_FREQ: 100
12+
13+
DATASET:
14+
COLOR_RGB: false
15+
DATASET: 'coco'
16+
ROOT: 'data/coco/'
17+
TEST_SET: 'val2017'
18+
TRAIN_SET: 'train2017'
19+
FLIP: true
20+
ROT_FACTOR: 40
21+
SCALE_FACTOR: 0.3
22+
MODEL:
23+
NAME: 'pose_resnet'
24+
PRETRAINED: 'models/pytorch/imagenet/resnet101-5d3b4d8f.pth'
25+
IMAGE_SIZE:
26+
- 192
27+
- 256
28+
HEATMAP_SIZE:
29+
- 48
30+
- 64
31+
SIGMA: 2
32+
NUM_JOINTS: 17
33+
TARGET_TYPE: 'gaussian'
34+
EXTRA:
35+
FINAL_CONV_KERNEL: 1
36+
DECONV_WITH_BIAS: false
37+
NUM_DECONV_LAYERS: 3
38+
NUM_DECONV_FILTERS:
39+
- 256
40+
- 256
41+
- 256
42+
NUM_DECONV_KERNELS:
43+
- 4
44+
- 4
45+
- 4
46+
NUM_LAYERS: 101
47+
LOSS:
48+
USE_TARGET_WEIGHT: true
49+
TRAIN:
50+
BATCH_SIZE_PER_GPU: 32
51+
SHUFFLE: true
52+
BEGIN_EPOCH: 0
53+
END_EPOCH: 140
54+
OPTIMIZER: 'adam'
55+
LR: 0.001
56+
LR_FACTOR: 0.1
57+
LR_STEP:
58+
- 90
59+
- 120
60+
WD: 0.0001
61+
GAMMA1: 0.99
62+
GAMMA2: 0.0
63+
MOMENTUM: 0.9
64+
NESTEROV: false
65+
TEST:
66+
BATCH_SIZE_PER_GPU: 32
67+
COCO_BBOX_FILE: 'data/coco/person_detection_results/COCO_val2017_detections_AP_H_56_person.json'
68+
BBOX_THRE: 1.0
69+
IMAGE_THRE: 0.0
70+
IN_VIS_THRE: 0.2
71+
MODEL_FILE: ''
72+
NMS_THRE: 1.0
73+
OKS_THRE: 0.9
74+
FLIP_TEST: true
75+
POST_PROCESS: true
76+
SHIFT_HEATMAP: true
77+
USE_GT_BBOX: true
78+
DEBUG:
79+
DEBUG: true
80+
SAVE_BATCH_IMAGES_GT: true
81+
SAVE_BATCH_IMAGES_PRED: true
82+
SAVE_HEATMAPS_GT: true
83+
SAVE_HEATMAPS_PRED: true

‎experiments/coco/resnet/res101_384x288_d256x3_adam_lr1e-3.yaml

+83
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,83 @@
1+
AUTO_RESUME: true
2+
CUDNN:
3+
BENCHMARK: true
4+
DETERMINISTIC: false
5+
ENABLED: true
6+
DATA_DIR: ''
7+
GPUS: (0,1,2,3)
8+
OUTPUT_DIR: 'output'
9+
LOG_DIR: 'log'
10+
WORKERS: 24
11+
PRINT_FREQ: 100
12+
13+
DATASET:
14+
COLOR_RGB: false
15+
DATASET: 'coco'
16+
ROOT: 'data/coco/'
17+
TEST_SET: 'val2017'
18+
TRAIN_SET: 'train2017'
19+
FLIP: true
20+
ROT_FACTOR: 40
21+
SCALE_FACTOR: 0.3
22+
MODEL:
23+
NAME: 'pose_resnet'
24+
PRETRAINED: 'models/pytorch/imagenet/resnet101-5d3b4d8f.pth'
25+
IMAGE_SIZE:
26+
- 288
27+
- 384
28+
HEATMAP_SIZE:
29+
- 72
30+
- 96
31+
SIGMA: 3
32+
NUM_JOINTS: 17
33+
TARGET_TYPE: 'gaussian'
34+
EXTRA:
35+
FINAL_CONV_KERNEL: 1
36+
DECONV_WITH_BIAS: false
37+
NUM_DECONV_LAYERS: 3
38+
NUM_DECONV_FILTERS:
39+
- 256
40+
- 256
41+
- 256
42+
NUM_DECONV_KERNELS:
43+
- 4
44+
- 4
45+
- 4
46+
NUM_LAYERS: 101
47+
LOSS:
48+
USE_TARGET_WEIGHT: true
49+
TRAIN:
50+
BATCH_SIZE_PER_GPU: 32
51+
SHUFFLE: true
52+
BEGIN_EPOCH: 0
53+
END_EPOCH: 140
54+
OPTIMIZER: 'adam'
55+
LR: 0.001
56+
LR_FACTOR: 0.1
57+
LR_STEP:
58+
- 90
59+
- 120
60+
WD: 0.0001
61+
GAMMA1: 0.99
62+
GAMMA2: 0.0
63+
MOMENTUM: 0.9
64+
NESTEROV: false
65+
TEST:
66+
BATCH_SIZE_PER_GPU: 32
67+
COCO_BBOX_FILE: 'data/coco/person_detection_results/COCO_val2017_detections_AP_H_56_person.json'
68+
BBOX_THRE: 1.0
69+
IMAGE_THRE: 0.0
70+
IN_VIS_THRE: 0.2
71+
MODEL_FILE: ''
72+
NMS_THRE: 1.0
73+
OKS_THRE: 0.9
74+
FLIP_TEST: true
75+
POST_PROCESS: true
76+
SHIFT_HEATMAP: true
77+
USE_GT_BBOX: true
78+
DEBUG:
79+
DEBUG: true
80+
SAVE_BATCH_IMAGES_GT: true
81+
SAVE_BATCH_IMAGES_PRED: true
82+
SAVE_HEATMAPS_GT: true
83+
SAVE_HEATMAPS_PRED: true

‎experiments/coco/resnet/res152_256x192_d256x3_adam_lr1e-3.yaml

+83
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,83 @@
1+
AUTO_RESUME: true
2+
CUDNN:
3+
BENCHMARK: true
4+
DETERMINISTIC: false
5+
ENABLED: true
6+
DATA_DIR: ''
7+
GPUS: (0,1,2,3)
8+
OUTPUT_DIR: 'output'
9+
LOG_DIR: 'log'
10+
WORKERS: 24
11+
PRINT_FREQ: 100
12+
13+
DATASET:
14+
COLOR_RGB: false
15+
DATASET: 'coco'
16+
ROOT: 'data/coco/'
17+
TEST_SET: 'val2017'
18+
TRAIN_SET: 'train2017'
19+
FLIP: true
20+
ROT_FACTOR: 40
21+
SCALE_FACTOR: 0.3
22+
MODEL:
23+
NAME: 'pose_resnet'
24+
PRETRAINED: 'models/pytorch/imagenet/resnet152-b121ed2d.pth'
25+
IMAGE_SIZE:
26+
- 192
27+
- 256
28+
HEATMAP_SIZE:
29+
- 48
30+
- 64
31+
SIGMA: 2
32+
NUM_JOINTS: 17
33+
TARGET_TYPE: 'gaussian'
34+
EXTRA:
35+
FINAL_CONV_KERNEL: 1
36+
DECONV_WITH_BIAS: false
37+
NUM_DECONV_LAYERS: 3
38+
NUM_DECONV_FILTERS:
39+
- 256
40+
- 256
41+
- 256
42+
NUM_DECONV_KERNELS:
43+
- 4
44+
- 4
45+
- 4
46+
NUM_LAYERS: 152
47+
LOSS:
48+
USE_TARGET_WEIGHT: true
49+
TRAIN:
50+
BATCH_SIZE_PER_GPU: 32
51+
SHUFFLE: true
52+
BEGIN_EPOCH: 0
53+
END_EPOCH: 140
54+
OPTIMIZER: 'adam'
55+
LR: 0.001
56+
LR_FACTOR: 0.1
57+
LR_STEP:
58+
- 90
59+
- 120
60+
WD: 0.0001
61+
GAMMA1: 0.99
62+
GAMMA2: 0.0
63+
MOMENTUM: 0.9
64+
NESTEROV: false
65+
TEST:
66+
BATCH_SIZE_PER_GPU: 32
67+
COCO_BBOX_FILE: 'data/coco/person_detection_results/COCO_val2017_detections_AP_H_56_person.json'
68+
BBOX_THRE: 1.0
69+
IMAGE_THRE: 0.0
70+
IN_VIS_THRE: 0.2
71+
MODEL_FILE: ''
72+
NMS_THRE: 1.0
73+
OKS_THRE: 0.9
74+
FLIP_TEST: true
75+
POST_PROCESS: true
76+
SHIFT_HEATMAP: true
77+
USE_GT_BBOX: true
78+
DEBUG:
79+
DEBUG: true
80+
SAVE_BATCH_IMAGES_GT: true
81+
SAVE_BATCH_IMAGES_PRED: true
82+
SAVE_HEATMAPS_GT: true
83+
SAVE_HEATMAPS_PRED: true

‎experiments/coco/resnet/res152_384x288_d256x3_adam_lr1e-3.yaml

+83
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,83 @@
1+
AUTO_RESUME: true
2+
CUDNN:
3+
BENCHMARK: true
4+
DETERMINISTIC: false
5+
ENABLED: true
6+
DATA_DIR: ''
7+
GPUS: (0,1,2,3)
8+
OUTPUT_DIR: 'output'
9+
LOG_DIR: 'log'
10+
WORKERS: 24
11+
PRINT_FREQ: 100
12+
13+
DATASET:
14+
COLOR_RGB: false
15+
DATASET: 'coco'
16+
ROOT: 'data/coco/'
17+
TEST_SET: 'val2017'
18+
TRAIN_SET: 'train2017'
19+
FLIP: true
20+
ROT_FACTOR: 40
21+
SCALE_FACTOR: 0.3
22+
MODEL:
23+
NAME: 'pose_resnet'
24+
PRETRAINED: 'models/pytorch/imagenet/resnet152-b121ed2d.pth'
25+
IMAGE_SIZE:
26+
- 288
27+
- 384
28+
HEATMAP_SIZE:
29+
- 72
30+
- 96
31+
SIGMA: 3
32+
NUM_JOINTS: 17
33+
TARGET_TYPE: 'gaussian'
34+
EXTRA:
35+
FINAL_CONV_KERNEL: 1
36+
DECONV_WITH_BIAS: false
37+
NUM_DECONV_LAYERS: 3
38+
NUM_DECONV_FILTERS:
39+
- 256
40+
- 256
41+
- 256
42+
NUM_DECONV_KERNELS:
43+
- 4
44+
- 4
45+
- 4
46+
NUM_LAYERS: 152
47+
LOSS:
48+
USE_TARGET_WEIGHT: true
49+
TRAIN:
50+
BATCH_SIZE_PER_GPU: 32
51+
SHUFFLE: true
52+
BEGIN_EPOCH: 0
53+
END_EPOCH: 140
54+
OPTIMIZER: 'adam'
55+
LR: 0.001
56+
LR_FACTOR: 0.1
57+
LR_STEP:
58+
- 90
59+
- 120
60+
WD: 0.0001
61+
GAMMA1: 0.99
62+
GAMMA2: 0.0
63+
MOMENTUM: 0.9
64+
NESTEROV: false
65+
TEST:
66+
BATCH_SIZE_PER_GPU: 32
67+
COCO_BBOX_FILE: 'data/coco/person_detection_results/COCO_val2017_detections_AP_H_56_person.json'
68+
BBOX_THRE: 1.0
69+
IMAGE_THRE: 0.0
70+
IN_VIS_THRE: 0.2
71+
MODEL_FILE: ''
72+
NMS_THRE: 1.0
73+
OKS_THRE: 0.9
74+
FLIP_TEST: true
75+
POST_PROCESS: true
76+
SHIFT_HEATMAP: true
77+
USE_GT_BBOX: true
78+
DEBUG:
79+
DEBUG: true
80+
SAVE_BATCH_IMAGES_GT: true
81+
SAVE_BATCH_IMAGES_PRED: true
82+
SAVE_HEATMAPS_GT: true
83+
SAVE_HEATMAPS_PRED: true

0 commit comments

Comments
 (0)