-
Notifications
You must be signed in to change notification settings - Fork 45
/
Copy pathmobilenetv2_512x512.yaml
146 lines (132 loc) · 2.68 KB
/
mobilenetv2_512x512.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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
SAMPLE_METHOD: 'coco_hp'
DATA_DIR: '/data'
EXP_ID: 'coco_pose_mobilenetv2'
DEBUG: 0
DEBUG_THEME: 'white'
SEED: 317
OUTPUT_DIR: '/data/centerpose/mobilenetv2'
LOG_DIR: ''
EXPERIMENT_NAME: ''
GPUS: [0, 1, 2, 3]
WORKERS: 8
PRINT_FREQ: 0
PIN_MEMORY: true
RANK: 0
SAVE_RESULTS: true
CUDNN:
BENCHMARK: true
MODEL:
INIT_WEIGHTS: false
PRETRAINED: '/data/pretrained_models/imagenet/hrnet_w32-36af842e.pth'
CENTER_THRESH: 0.1
NUM_CLASSES: 1
NAME: 'mobilenetv2'
HEADS_NAME: 'keypoint'
HEADS_NUM: [1, 2, 34, 2, 17, 2]
HEAD_CONV: 256
INTERMEDIATE_CHANNEL: 24
DOWN_RATIO: 4
NUM_STACKS: 1
INPUT_RES: 512
OUTPUT_RES: 128
INPUT_H: 512
INPUT_W: 512
PAD: 31
NUM_KEYPOINTS: 17
TAG_PER_JOINT: true
TARGET_TYPE: 'gaussian'
SIGMA: 2
LOSS:
METRIC: 'loss'
MSE_LOSS: false
REG_LOSS: 'l1'
USE_OHKM: false
TOPK: 8
USE_TARGET_WEIGHT: true
USE_DIFFERENT_JOINTS_WEIGHT: false
HP_WEIGHT: 1.
HM_HP_WEIGHT: 1.
DENSE_HP: false
HM_HP: true
REG_BBOX: true
WH_WEIGHT: 0.1
REG_OFFSET: true
OFF_WEIGHT: 1.
REG_HP_OFFSET: true
HM_HP_WEIGHT: 1.
DATASET:
DATASET: 'coco_hp'
TRAIN_SET: 'train'
TEST_SET: 'valid'
TRAIN_IMAGE_DIR: 'images/train2017'
TRAIN_ANNOTATIONS: ['person_keypoints_train2017.json']
VAL_IMAGE_DIR: 'images/val2017'
VAL_ANNOTATIONS: 'person_keypoints_val2017.json'
# training data augmentation
MEAN: [0.408, 0.447, 0.470]
STD: [0.289, 0.274, 0.278]
SHIFT: 0.1
SCALE: 0.4
ROTATE: 0.
# for pose
AUG_ROT: 0.
FLIP: 0.5
NO_COLOR_AUG: false
ROT_FACTOR: 30
SCALE_MIN: 0.5
SCALE_MAX: 1.1
IMAGE_SIZE: 512
RANDOM_CROP: true
TRAIN:
OPTIMIZER: 'adam'
DISTRIBUTE: true
LOCAL_RANK: 0
HIDE_DATA_TIME: false
SAVE_ALL_MODEL: false
RESUME: false
LR_FACTOR: 0.1
LR_STEP: [270, 300]
EPOCHS: 320
NUM_ITERS: -1
LR: 5.e-4
BATCH_SIZE: 128
MASTER_BATCH_SIZE: 32
MOMENTUM: 0.9
WD: 0.0001
NESTEROV: false
GAMMA1: 0.99
GAMMA2: 0.0
# 'apply and reset gradients every n batches'
STRIDE_APPLY: 1
CHECKPOINT: ''
SHUFFLE: true
VAL_INTERVALS: 1
TRAINVAL: false
TEST:
# Test Model Epoch
MODEL_PATH: '/home/tensorboy/data/centerpose/model_zoo/mobilenetV3_1x.pth'
TASK: 'multi_pose'
FLIP_TEST: true
DEMO_FILE: '../images/33823288584_1d21cf0a26_k.jpg'
MODEL_FILE: ''
TEST_SCALES: [1]
IMAGE_THRE: 0.1
TOPK: 100
NMS: false
NMS_THRE: 0.5
NOT_PREFETCH_TEST: false
FIX_RES: false
KEEP_RES: false
SOFT_NMS: false
OKS_THRE: 0.5
VIS_THRESH: 0.3
KEYPOINT_THRESH: 0.2
NUM_MIN_KPT: 4
THRESH_HUMAN: 0.4
EVAL_ORACLE_HM: false
EVAL_ORACLE_WH: false
EVAL_ORACLE_OFFSET: false
EVAL_ORACLE_KPS: false
EVAL_ORACLE_HMHP: false
EVAL_ORACLE_HP_OFFSET: false
EVAL_ORACLE_DEP: false