| DATASET: | |
| TRAIN: | |
| TYPE: 'Mscoco' | |
| ROOT: './data/coco/' | |
| IMG_PREFIX: 'train2017' | |
| ANN: 'annotations/person_keypoints_train2017.json' | |
| AUG: | |
| FLIP: true | |
| ROT_FACTOR: 40 | |
| SCALE_FACTOR: 0.3 | |
| NUM_JOINTS_HALF_BODY: 8 | |
| PROB_HALF_BODY: -1 | |
| VAL: | |
| TYPE: 'Mscoco' | |
| ROOT: './data/coco/' | |
| IMG_PREFIX: 'val2017' | |
| ANN: 'annotations/person_keypoints_val2017.json' | |
| TEST: | |
| TYPE: 'Mscoco_det' | |
| ROOT: './data/coco/' | |
| IMG_PREFIX: 'val2017' | |
| DET_FILE: './exp/json/test_det_yolo.json' | |
| ANN: 'annotations/person_keypoints_val2017.json' | |
| DATA_PRESET: | |
| TYPE: 'simple' | |
| SIGMA: 2 | |
| NUM_JOINTS: 17 | |
| IMAGE_SIZE: | |
| - 256 | |
| - 192 | |
| HEATMAP_SIZE: | |
| - 64 | |
| - 48 | |
| MODEL: | |
| TYPE: 'FastPose_DUC' | |
| BACKBONE: 'se-resnet' | |
| PRETRAINED: '' | |
| TRY_LOAD: '' | |
| NUM_DECONV_FILTERS: | |
| - 256 | |
| - 256 | |
| - 256 | |
| NUM_LAYERS: 152 | |
| FINAL_CONV_KERNEL: 1 | |
| STAGE1: | |
| NUM_CONV: 4 | |
| STAGE2: | |
| NUM_CONV: 2 | |
| STAGE3: | |
| NUM_CONV: 1 | |
| LOSS: | |
| TYPE: 'MSELoss' | |
| DETECTOR: | |
| NAME: 'yolo' | |
| CONFIG: 'detector/yolo/cfg/yolov3-spp.cfg' | |
| WEIGHTS: 'detector/yolo/data/yolov3-spp.weights' | |
| NMS_THRES: 0.6 | |
| CONFIDENCE: 0.05 | |
| TRAIN: | |
| WORLD_SIZE: 4 | |
| BATCH_SIZE: 32 | |
| BEGIN_EPOCH: 0 | |
| END_EPOCH: 200 | |
| OPTIMIZER: 'adam' | |
| LR: 0.001 | |
| LR_FACTOR: 0.1 | |
| LR_STEP: | |
| - 90 | |
| - 120 | |
| DPG_MILESTONE: 140 | |
| DPG_STEP: | |
| - 160 | |
| - 190 | |