removed configuration files for philly
This commit is contained in:
Родитель
d128ae6479
Коммит
04aca888eb
|
@ -1,84 +0,0 @@
|
||||||
GPUS: '0'
|
|
||||||
BACKBONE_MODEL: pose_resnet_antialiased
|
|
||||||
MODEL: multiview_pose_resnet_antialiased
|
|
||||||
DATA_DIR: ''
|
|
||||||
OUTPUT_DIR: 'output'
|
|
||||||
LOG_DIR: 'log'
|
|
||||||
WORKERS: 4
|
|
||||||
PRINT_FREQ: 100
|
|
||||||
|
|
||||||
DATASET:
|
|
||||||
TRAIN_DATASET: mixed
|
|
||||||
TEST_DATASET: multiview_h36m
|
|
||||||
ROOT: 'data/'
|
|
||||||
TRAIN_SUBSET: train
|
|
||||||
TEST_SUBSET: validation
|
|
||||||
DATA_FORMAT: 'zip'
|
|
||||||
ROT_FACTOR: 0
|
|
||||||
SCALE_FACTOR: 0
|
|
||||||
NETWORK:
|
|
||||||
PRETRAINED: 'models/pytorch/imagenet/resnet50_lpf3.pth.tar'
|
|
||||||
TARGET_TYPE: gaussian
|
|
||||||
AGGRE: true
|
|
||||||
HEATMAP_SIZE:
|
|
||||||
- 64
|
|
||||||
- 64
|
|
||||||
IMAGE_SIZE:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_JOINTS: 20
|
|
||||||
SIGMA: 2
|
|
||||||
POSE_RESNET:
|
|
||||||
FINAL_CONV_KERNEL: 1
|
|
||||||
DECONV_WITH_BIAS: false
|
|
||||||
NUM_DECONV_LAYERS: 3
|
|
||||||
NUM_DECONV_FILTERS:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_DECONV_KERNELS:
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
NUM_LAYERS: 50
|
|
||||||
LOSS:
|
|
||||||
USE_TARGET_WEIGHT: true
|
|
||||||
TRAIN:
|
|
||||||
BATCH_SIZE: 8
|
|
||||||
SHUFFLE: true
|
|
||||||
BEGIN_EPOCH: 0
|
|
||||||
END_EPOCH: 30
|
|
||||||
RESUME: true
|
|
||||||
OPTIMIZER: adam
|
|
||||||
LR: 0.001
|
|
||||||
LR_FACTOR: 0.1
|
|
||||||
LR_STEP:
|
|
||||||
- 20
|
|
||||||
- 25
|
|
||||||
WD: 0.0001
|
|
||||||
GAMMA1: 0.99
|
|
||||||
GAMMA2: 0.0
|
|
||||||
MOMENTUM: 0.9
|
|
||||||
NESTEROV: false
|
|
||||||
TEST:
|
|
||||||
BATCH_SIZE: 4
|
|
||||||
BBOX_FILE: ''
|
|
||||||
BBOX_THRE: 1.0
|
|
||||||
DETECTOR: fpn_dcn
|
|
||||||
DETECTOR_DIR: ''
|
|
||||||
IMAGE_THRE: 0.1
|
|
||||||
IN_VIS_THRE: 0.0
|
|
||||||
MATCH_IOU_THRE: 0.3
|
|
||||||
MODEL_FILE: ''
|
|
||||||
NMS_THRE: 0.6
|
|
||||||
OKS_THRE: 0.5
|
|
||||||
POST_PROCESS: false
|
|
||||||
USE_GT_BBOX: true
|
|
||||||
PICT_STRUCT:
|
|
||||||
PAIRWISE_FILE: 'data/pict/pairwise.pkl'
|
|
||||||
DEBUG:
|
|
||||||
DEBUG: true
|
|
||||||
SAVE_BATCH_IMAGES_GT: true
|
|
||||||
SAVE_BATCH_IMAGES_PRED: true
|
|
||||||
SAVE_HEATMAPS_GT: true
|
|
||||||
SAVE_HEATMAPS_PRED: true
|
|
|
@ -1,84 +0,0 @@
|
||||||
GPUS: '0'
|
|
||||||
BACKBONE_MODEL: pose_resnet
|
|
||||||
MODEL: multiview_pose_resnet
|
|
||||||
DATA_DIR: ''
|
|
||||||
OUTPUT_DIR: 'output'
|
|
||||||
LOG_DIR: 'log'
|
|
||||||
WORKERS: 8
|
|
||||||
PRINT_FREQ: 100
|
|
||||||
|
|
||||||
DATASET:
|
|
||||||
TRAIN_DATASET: mixed
|
|
||||||
TRAIN_SUBSET: train
|
|
||||||
TEST_DATASET: multiview_h36m
|
|
||||||
TEST_SUBSET: validation
|
|
||||||
ROOT: 'data/'
|
|
||||||
DATA_FORMAT: 'jpg'
|
|
||||||
ROT_FACTOR: 0
|
|
||||||
SCALE_FACTOR: 0
|
|
||||||
ROOTIDX: 0
|
|
||||||
NETWORK:
|
|
||||||
PRETRAINED: 'models/pytorch/imagenet/resnet152-b121ed2d.pth'
|
|
||||||
TARGET_TYPE: gaussian
|
|
||||||
AGGRE: false
|
|
||||||
HEATMAP_SIZE:
|
|
||||||
- 64
|
|
||||||
- 64
|
|
||||||
IMAGE_SIZE:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_JOINTS: 20
|
|
||||||
SIGMA: 2
|
|
||||||
POSE_RESNET:
|
|
||||||
FINAL_CONV_KERNEL: 1
|
|
||||||
DECONV_WITH_BIAS: false
|
|
||||||
NUM_DECONV_LAYERS: 3
|
|
||||||
NUM_DECONV_FILTERS:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_DECONV_KERNELS:
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
NUM_LAYERS: 152
|
|
||||||
LOSS:
|
|
||||||
USE_TARGET_WEIGHT: true
|
|
||||||
TRAIN:
|
|
||||||
BATCH_SIZE: 8
|
|
||||||
SHUFFLE: true
|
|
||||||
MODEL_FILE: 'models/pytorch/multiview_pose_resnet/multiview_pose_resnet152_mixed_noaggre_best.pth.tar'
|
|
||||||
BEGIN_EPOCH: 0
|
|
||||||
END_EPOCH: 30
|
|
||||||
RESUME: false
|
|
||||||
OPTIMIZER: adam
|
|
||||||
LR: 0.001
|
|
||||||
LR_FACTOR: 0.1
|
|
||||||
LR_STEP:
|
|
||||||
- 20
|
|
||||||
- 25
|
|
||||||
WD: 0.0001
|
|
||||||
GAMMA1: 0.99
|
|
||||||
GAMMA2: 0.0
|
|
||||||
MOMENTUM: 0.9
|
|
||||||
NESTEROV: false
|
|
||||||
TEST:
|
|
||||||
BATCH_SIZE: 8
|
|
||||||
BBOX_FILE: ''
|
|
||||||
BBOX_THRE: 1.0
|
|
||||||
DETECTOR: fpn_dcn
|
|
||||||
DETECTOR_DIR: ''
|
|
||||||
IMAGE_THRE: 0.1
|
|
||||||
IN_VIS_THRE: 0.0
|
|
||||||
MATCH_IOU_THRE: 0.3
|
|
||||||
MODEL_FILE: ''
|
|
||||||
NMS_THRE: 0.6
|
|
||||||
OKS_THRE: 0.5
|
|
||||||
POST_PROCESS: false
|
|
||||||
USE_GT_BBOX: true
|
|
||||||
DEBUG:
|
|
||||||
DEBUG: true
|
|
||||||
SAVE_BATCH_IMAGES_GT: true
|
|
||||||
SAVE_BATCH_IMAGES_PRED: true
|
|
||||||
SAVE_HEATMAPS_GT: true
|
|
||||||
SAVE_HEATMAPS_PRED: true
|
|
|
@ -1,85 +0,0 @@
|
||||||
GPUS: '0,1,2,3'
|
|
||||||
BACKBONE_MODEL: pose_hrnet
|
|
||||||
MODEL: multiview_pose_hrnet
|
|
||||||
DATA_DIR: ''
|
|
||||||
OUTPUT_DIR: 'output'
|
|
||||||
LOG_DIR: 'log'
|
|
||||||
WORKERS: 16
|
|
||||||
PRINT_FREQ: 100
|
|
||||||
|
|
||||||
DATASET:
|
|
||||||
TRAIN_DATASET: mixed
|
|
||||||
TEST_DATASET: multiview_h36m
|
|
||||||
ROOT: '../mixed_data/'
|
|
||||||
TRAIN_SUBSET: train
|
|
||||||
TEST_SUBSET: validation
|
|
||||||
DATA_FORMAT: 'zip'
|
|
||||||
ROT_FACTOR: 0
|
|
||||||
SCALE_FACTOR: 0
|
|
||||||
NETWORK:
|
|
||||||
PRETRAINED: 'models/pytorch/imagenet/hrnet_w48-8ef0771d.pth'
|
|
||||||
TARGET_TYPE: gaussian
|
|
||||||
AGGRE: true
|
|
||||||
HEATMAP_SIZE:
|
|
||||||
- 64
|
|
||||||
- 64
|
|
||||||
IMAGE_SIZE:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_JOINTS: 20
|
|
||||||
SIGMA: 2
|
|
||||||
POSE_RESNET:
|
|
||||||
FINAL_CONV_KERNEL: 1
|
|
||||||
DECONV_WITH_BIAS: false
|
|
||||||
NUM_DECONV_LAYERS: 3
|
|
||||||
NUM_DECONV_FILTERS:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_DECONV_KERNELS:
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
NUM_LAYERS: 50
|
|
||||||
LOSS:
|
|
||||||
USE_TARGET_WEIGHT: true
|
|
||||||
TRAIN:
|
|
||||||
BATCH_SIZE: 8
|
|
||||||
SHUFFLE: true
|
|
||||||
BEGIN_EPOCH: 0
|
|
||||||
END_EPOCH: 30
|
|
||||||
RESUME: true
|
|
||||||
OPTIMIZER: adam
|
|
||||||
LR: 0.001
|
|
||||||
LR_FACTOR: 0.1
|
|
||||||
LR_STEP:
|
|
||||||
- 20
|
|
||||||
- 25
|
|
||||||
WD: 0.0001
|
|
||||||
GAMMA1: 0.99
|
|
||||||
GAMMA2: 0.0
|
|
||||||
MOMENTUM: 0.9
|
|
||||||
NESTEROV: false
|
|
||||||
TEST:
|
|
||||||
BATCH_SIZE: 4
|
|
||||||
BBOX_FILE: ''
|
|
||||||
BBOX_THRE: 1.0
|
|
||||||
DETECTOR: fpn_dcn
|
|
||||||
DETECTOR_DIR: ''
|
|
||||||
IMAGE_THRE: 0.1
|
|
||||||
IN_VIS_THRE: 0.0
|
|
||||||
MATCH_IOU_THRE: 0.3
|
|
||||||
MODEL_FILE: ''
|
|
||||||
NMS_THRE: 0.6
|
|
||||||
OKS_THRE: 0.5
|
|
||||||
POST_PROCESS: false
|
|
||||||
USE_GT_BBOX: true
|
|
||||||
PICT_STRUCT:
|
|
||||||
FIRST_NBINS: 16
|
|
||||||
PAIRWISE_FILE: 'data/pict/pairwise.pkl'
|
|
||||||
DEBUG:
|
|
||||||
DEBUG: true
|
|
||||||
SAVE_BATCH_IMAGES_GT: true
|
|
||||||
SAVE_BATCH_IMAGES_PRED: true
|
|
||||||
SAVE_HEATMAPS_GT: true
|
|
||||||
SAVE_HEATMAPS_PRED: true
|
|
|
@ -1,86 +0,0 @@
|
||||||
GPUS: '0,1,2,3'
|
|
||||||
BACKBONE_MODEL: pose_hrnet
|
|
||||||
MODEL: multiview_pose_hrnet
|
|
||||||
DATA_DIR: ''
|
|
||||||
OUTPUT_DIR: 'output'
|
|
||||||
LOG_DIR: 'log'
|
|
||||||
WORKERS: 16
|
|
||||||
PRINT_FREQ: 100
|
|
||||||
|
|
||||||
DATASET:
|
|
||||||
TRAIN_DATASET: mixed
|
|
||||||
TEST_DATASET: multiview_h36m
|
|
||||||
ROOT: '../mixed_data'
|
|
||||||
TRAIN_SUBSET: train
|
|
||||||
TEST_SUBSET: validation
|
|
||||||
DATA_FORMAT: 'zip'
|
|
||||||
ROT_FACTOR: 0
|
|
||||||
SCALE_FACTOR: 0
|
|
||||||
CROP: False
|
|
||||||
NETWORK:
|
|
||||||
PRETRAINED: 'models/pytorch/imagenet/hrnet_w48-8ef0771d.pth'
|
|
||||||
TARGET_TYPE: gaussian
|
|
||||||
AGGRE: true
|
|
||||||
HEATMAP_SIZE:
|
|
||||||
- 64
|
|
||||||
- 64
|
|
||||||
IMAGE_SIZE:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_JOINTS: 20
|
|
||||||
SIGMA: 2
|
|
||||||
POSE_RESNET:
|
|
||||||
FINAL_CONV_KERNEL: 1
|
|
||||||
DECONV_WITH_BIAS: false
|
|
||||||
NUM_DECONV_LAYERS: 3
|
|
||||||
NUM_DECONV_FILTERS:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_DECONV_KERNELS:
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
NUM_LAYERS: 50
|
|
||||||
LOSS:
|
|
||||||
USE_TARGET_WEIGHT: true
|
|
||||||
TRAIN:
|
|
||||||
BATCH_SIZE: 8
|
|
||||||
SHUFFLE: true
|
|
||||||
BEGIN_EPOCH: 0
|
|
||||||
END_EPOCH: 30
|
|
||||||
RESUME: true
|
|
||||||
OPTIMIZER: adam
|
|
||||||
LR: 0.001
|
|
||||||
LR_FACTOR: 0.1
|
|
||||||
LR_STEP:
|
|
||||||
- 20
|
|
||||||
- 25
|
|
||||||
WD: 0.0001
|
|
||||||
GAMMA1: 0.99
|
|
||||||
GAMMA2: 0.0
|
|
||||||
MOMENTUM: 0.9
|
|
||||||
NESTEROV: false
|
|
||||||
TEST:
|
|
||||||
BATCH_SIZE: 4
|
|
||||||
BBOX_FILE: ''
|
|
||||||
BBOX_THRE: 1.0
|
|
||||||
DETECTOR: fpn_dcn
|
|
||||||
DETECTOR_DIR: ''
|
|
||||||
IMAGE_THRE: 0.1
|
|
||||||
IN_VIS_THRE: 0.0
|
|
||||||
MATCH_IOU_THRE: 0.3
|
|
||||||
MODEL_FILE: ''
|
|
||||||
NMS_THRE: 0.6
|
|
||||||
OKS_THRE: 0.5
|
|
||||||
POST_PROCESS: false
|
|
||||||
USE_GT_BBOX: true
|
|
||||||
PICT_STRUCT:
|
|
||||||
FIRST_NBINS: 16
|
|
||||||
PAIRWISE_FILE: 'data/pict/pairwise.pkl'
|
|
||||||
DEBUG:
|
|
||||||
DEBUG: true
|
|
||||||
SAVE_BATCH_IMAGES_GT: true
|
|
||||||
SAVE_BATCH_IMAGES_PRED: true
|
|
||||||
SAVE_HEATMAPS_GT: true
|
|
||||||
SAVE_HEATMAPS_PRED: true
|
|
|
@ -1,82 +0,0 @@
|
||||||
GPUS: '0,1,2,3'
|
|
||||||
BACKBONE_MODEL: pose_mobilenetv2
|
|
||||||
MODEL: multiview_pose_mobilenetv2
|
|
||||||
DATA_DIR: ''
|
|
||||||
OUTPUT_DIR: 'output'
|
|
||||||
LOG_DIR: 'log'
|
|
||||||
WORKERS: 16
|
|
||||||
PRINT_FREQ: 100
|
|
||||||
|
|
||||||
DATASET:
|
|
||||||
TRAIN_DATASET: mixed
|
|
||||||
TEST_DATASET: multiview_h36m
|
|
||||||
ROOT: '../mixed_data/'
|
|
||||||
TRAIN_SUBSET: train
|
|
||||||
TEST_SUBSET: validation
|
|
||||||
DATA_FORMAT: 'zip'
|
|
||||||
ROT_FACTOR: 0
|
|
||||||
SCALE_FACTOR: 0
|
|
||||||
NETWORK:
|
|
||||||
PRETRAINED: 'models/pytorch/imagenet/mobilenet_v2.pth.tar'
|
|
||||||
TARGET_TYPE: gaussian
|
|
||||||
AGGRE: true
|
|
||||||
HEATMAP_SIZE:
|
|
||||||
- 64
|
|
||||||
- 64
|
|
||||||
IMAGE_SIZE:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_JOINTS: 20
|
|
||||||
SIGMA: 2
|
|
||||||
POSE_RESNET:
|
|
||||||
FINAL_CONV_KERNEL: 1
|
|
||||||
DECONV_WITH_BIAS: false
|
|
||||||
NUM_DECONV_LAYERS: 3
|
|
||||||
NUM_DECONV_FILTERS:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_DECONV_KERNELS:
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
NUM_LAYERS: 50
|
|
||||||
LOSS:
|
|
||||||
USE_TARGET_WEIGHT: true
|
|
||||||
TRAIN:
|
|
||||||
BATCH_SIZE: 8
|
|
||||||
SHUFFLE: true
|
|
||||||
BEGIN_EPOCH: 0
|
|
||||||
END_EPOCH: 30
|
|
||||||
RESUME: true
|
|
||||||
OPTIMIZER: adam
|
|
||||||
LR: 0.001
|
|
||||||
LR_FACTOR: 0.1
|
|
||||||
LR_STEP:
|
|
||||||
- 20
|
|
||||||
- 25
|
|
||||||
WD: 0.0001
|
|
||||||
GAMMA1: 0.99
|
|
||||||
GAMMA2: 0.0
|
|
||||||
MOMENTUM: 0.9
|
|
||||||
NESTEROV: false
|
|
||||||
TEST:
|
|
||||||
BATCH_SIZE: 4
|
|
||||||
BBOX_FILE: ''
|
|
||||||
BBOX_THRE: 1.0
|
|
||||||
DETECTOR: fpn_dcn
|
|
||||||
DETECTOR_DIR: ''
|
|
||||||
IMAGE_THRE: 0.1
|
|
||||||
IN_VIS_THRE: 0.0
|
|
||||||
MATCH_IOU_THRE: 0.3
|
|
||||||
MODEL_FILE: ''
|
|
||||||
NMS_THRE: 0.6
|
|
||||||
OKS_THRE: 0.5
|
|
||||||
POST_PROCESS: false
|
|
||||||
USE_GT_BBOX: true
|
|
||||||
DEBUG:
|
|
||||||
DEBUG: true
|
|
||||||
SAVE_BATCH_IMAGES_GT: true
|
|
||||||
SAVE_BATCH_IMAGES_PRED: true
|
|
||||||
SAVE_HEATMAPS_GT: true
|
|
||||||
SAVE_HEATMAPS_PRED: true
|
|
|
@ -1,82 +0,0 @@
|
||||||
GPUS: '0,1,2,3'
|
|
||||||
BACKBONE_MODEL: pose_mobilenetv2
|
|
||||||
MODEL: multiview_pose_mobilenetv2
|
|
||||||
DATA_DIR: ''
|
|
||||||
OUTPUT_DIR: 'output'
|
|
||||||
LOG_DIR: 'log'
|
|
||||||
WORKERS: 16
|
|
||||||
PRINT_FREQ: 100
|
|
||||||
|
|
||||||
DATASET:
|
|
||||||
TRAIN_DATASET: mixed
|
|
||||||
TEST_DATASET: multiview_h36m
|
|
||||||
ROOT: '../mixed_data/'
|
|
||||||
TRAIN_SUBSET: train
|
|
||||||
TEST_SUBSET: validation
|
|
||||||
DATA_FORMAT: 'zip'
|
|
||||||
ROT_FACTOR: 0
|
|
||||||
SCALE_FACTOR: 0
|
|
||||||
NETWORK:
|
|
||||||
PRETRAINED: 'models/pytorch/imagenet/mobilenet_v2.pth.tar'
|
|
||||||
TARGET_TYPE: gaussian
|
|
||||||
AGGRE: false
|
|
||||||
HEATMAP_SIZE:
|
|
||||||
- 64
|
|
||||||
- 64
|
|
||||||
IMAGE_SIZE:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_JOINTS: 20
|
|
||||||
SIGMA: 2
|
|
||||||
POSE_RESNET:
|
|
||||||
FINAL_CONV_KERNEL: 1
|
|
||||||
DECONV_WITH_BIAS: false
|
|
||||||
NUM_DECONV_LAYERS: 3
|
|
||||||
NUM_DECONV_FILTERS:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_DECONV_KERNELS:
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
NUM_LAYERS: 50
|
|
||||||
LOSS:
|
|
||||||
USE_TARGET_WEIGHT: true
|
|
||||||
TRAIN:
|
|
||||||
BATCH_SIZE: 8
|
|
||||||
SHUFFLE: true
|
|
||||||
BEGIN_EPOCH: 0
|
|
||||||
END_EPOCH: 30
|
|
||||||
RESUME: true
|
|
||||||
OPTIMIZER: adam
|
|
||||||
LR: 0.001
|
|
||||||
LR_FACTOR: 0.1
|
|
||||||
LR_STEP:
|
|
||||||
- 20
|
|
||||||
- 25
|
|
||||||
WD: 0.0001
|
|
||||||
GAMMA1: 0.99
|
|
||||||
GAMMA2: 0.0
|
|
||||||
MOMENTUM: 0.9
|
|
||||||
NESTEROV: false
|
|
||||||
TEST:
|
|
||||||
BATCH_SIZE: 4
|
|
||||||
BBOX_FILE: ''
|
|
||||||
BBOX_THRE: 1.0
|
|
||||||
DETECTOR: fpn_dcn
|
|
||||||
DETECTOR_DIR: ''
|
|
||||||
IMAGE_THRE: 0.1
|
|
||||||
IN_VIS_THRE: 0.0
|
|
||||||
MATCH_IOU_THRE: 0.3
|
|
||||||
MODEL_FILE: ''
|
|
||||||
NMS_THRE: 0.6
|
|
||||||
OKS_THRE: 0.5
|
|
||||||
POST_PROCESS: false
|
|
||||||
USE_GT_BBOX: true
|
|
||||||
DEBUG:
|
|
||||||
DEBUG: true
|
|
||||||
SAVE_BATCH_IMAGES_GT: true
|
|
||||||
SAVE_BATCH_IMAGES_PRED: true
|
|
||||||
SAVE_HEATMAPS_GT: true
|
|
||||||
SAVE_HEATMAPS_PRED: true
|
|
|
@ -1,82 +0,0 @@
|
||||||
GPUS: '0,1,2,3'
|
|
||||||
BACKBONE_MODEL: pose_mobilenetv2
|
|
||||||
MODEL: multiview_pose_mobilenetv2
|
|
||||||
DATA_DIR: ''
|
|
||||||
OUTPUT_DIR: 'output'
|
|
||||||
LOG_DIR: 'log'
|
|
||||||
WORKERS: 16
|
|
||||||
PRINT_FREQ: 100
|
|
||||||
|
|
||||||
DATASET:
|
|
||||||
TRAIN_DATASET: mixed
|
|
||||||
TEST_DATASET: multiview_h36m
|
|
||||||
ROOT: '../mixed_data/'
|
|
||||||
TRAIN_SUBSET: train
|
|
||||||
TEST_SUBSET: validation
|
|
||||||
DATA_FORMAT: 'zip'
|
|
||||||
ROT_FACTOR: 0
|
|
||||||
SCALE_FACTOR: 0
|
|
||||||
NETWORK:
|
|
||||||
PRETRAINED: 'models/pytorch/imagenet/mobilenet_v2.pth.tar'
|
|
||||||
TARGET_TYPE: gaussian
|
|
||||||
AGGRE: true
|
|
||||||
HEATMAP_SIZE:
|
|
||||||
- 80
|
|
||||||
- 80
|
|
||||||
IMAGE_SIZE:
|
|
||||||
- 320
|
|
||||||
- 320
|
|
||||||
NUM_JOINTS: 20
|
|
||||||
SIGMA: 3
|
|
||||||
POSE_RESNET:
|
|
||||||
FINAL_CONV_KERNEL: 1
|
|
||||||
DECONV_WITH_BIAS: false
|
|
||||||
NUM_DECONV_LAYERS: 3
|
|
||||||
NUM_DECONV_FILTERS:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_DECONV_KERNELS:
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
NUM_LAYERS: 50
|
|
||||||
LOSS:
|
|
||||||
USE_TARGET_WEIGHT: true
|
|
||||||
TRAIN:
|
|
||||||
BATCH_SIZE: 4
|
|
||||||
SHUFFLE: true
|
|
||||||
BEGIN_EPOCH: 0
|
|
||||||
END_EPOCH: 30
|
|
||||||
RESUME: true
|
|
||||||
OPTIMIZER: adam
|
|
||||||
LR: 0.001
|
|
||||||
LR_FACTOR: 0.1
|
|
||||||
LR_STEP:
|
|
||||||
- 20
|
|
||||||
- 25
|
|
||||||
WD: 0.0001
|
|
||||||
GAMMA1: 0.99
|
|
||||||
GAMMA2: 0.0
|
|
||||||
MOMENTUM: 0.9
|
|
||||||
NESTEROV: false
|
|
||||||
TEST:
|
|
||||||
BATCH_SIZE: 4
|
|
||||||
BBOX_FILE: ''
|
|
||||||
BBOX_THRE: 1.0
|
|
||||||
DETECTOR: fpn_dcn
|
|
||||||
DETECTOR_DIR: ''
|
|
||||||
IMAGE_THRE: 0.1
|
|
||||||
IN_VIS_THRE: 0.0
|
|
||||||
MATCH_IOU_THRE: 0.3
|
|
||||||
MODEL_FILE: ''
|
|
||||||
NMS_THRE: 0.6
|
|
||||||
OKS_THRE: 0.5
|
|
||||||
POST_PROCESS: false
|
|
||||||
USE_GT_BBOX: true
|
|
||||||
DEBUG:
|
|
||||||
DEBUG: true
|
|
||||||
SAVE_BATCH_IMAGES_GT: true
|
|
||||||
SAVE_BATCH_IMAGES_PRED: true
|
|
||||||
SAVE_HEATMAPS_GT: true
|
|
||||||
SAVE_HEATMAPS_PRED: true
|
|
|
@ -1,82 +0,0 @@
|
||||||
GPUS: '0,1,2,3'
|
|
||||||
BACKBONE_MODEL: pose_mobilenetv2
|
|
||||||
MODEL: multiview_pose_mobilenetv2
|
|
||||||
DATA_DIR: ''
|
|
||||||
OUTPUT_DIR: 'output'
|
|
||||||
LOG_DIR: 'log'
|
|
||||||
WORKERS: 16
|
|
||||||
PRINT_FREQ: 100
|
|
||||||
|
|
||||||
DATASET:
|
|
||||||
TRAIN_DATASET: mixed
|
|
||||||
TEST_DATASET: multiview_h36m
|
|
||||||
ROOT: '../mixed_data/'
|
|
||||||
TRAIN_SUBSET: train
|
|
||||||
TEST_SUBSET: validation
|
|
||||||
DATA_FORMAT: 'zip'
|
|
||||||
ROT_FACTOR: 0
|
|
||||||
SCALE_FACTOR: 0
|
|
||||||
NETWORK:
|
|
||||||
PRETRAINED: 'models/pytorch/imagenet/mobilenet_v2.pth.tar'
|
|
||||||
TARGET_TYPE: gaussian
|
|
||||||
AGGRE: false
|
|
||||||
HEATMAP_SIZE:
|
|
||||||
- 80
|
|
||||||
- 80
|
|
||||||
IMAGE_SIZE:
|
|
||||||
- 320
|
|
||||||
- 320
|
|
||||||
NUM_JOINTS: 20
|
|
||||||
SIGMA: 3
|
|
||||||
POSE_RESNET:
|
|
||||||
FINAL_CONV_KERNEL: 1
|
|
||||||
DECONV_WITH_BIAS: false
|
|
||||||
NUM_DECONV_LAYERS: 3
|
|
||||||
NUM_DECONV_FILTERS:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_DECONV_KERNELS:
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
NUM_LAYERS: 50
|
|
||||||
LOSS:
|
|
||||||
USE_TARGET_WEIGHT: true
|
|
||||||
TRAIN:
|
|
||||||
BATCH_SIZE: 4
|
|
||||||
SHUFFLE: true
|
|
||||||
BEGIN_EPOCH: 0
|
|
||||||
END_EPOCH: 30
|
|
||||||
RESUME: true
|
|
||||||
OPTIMIZER: adam
|
|
||||||
LR: 0.001
|
|
||||||
LR_FACTOR: 0.1
|
|
||||||
LR_STEP:
|
|
||||||
- 20
|
|
||||||
- 25
|
|
||||||
WD: 0.0001
|
|
||||||
GAMMA1: 0.99
|
|
||||||
GAMMA2: 0.0
|
|
||||||
MOMENTUM: 0.9
|
|
||||||
NESTEROV: false
|
|
||||||
TEST:
|
|
||||||
BATCH_SIZE: 4
|
|
||||||
BBOX_FILE: ''
|
|
||||||
BBOX_THRE: 1.0
|
|
||||||
DETECTOR: fpn_dcn
|
|
||||||
DETECTOR_DIR: ''
|
|
||||||
IMAGE_THRE: 0.1
|
|
||||||
IN_VIS_THRE: 0.0
|
|
||||||
MATCH_IOU_THRE: 0.3
|
|
||||||
MODEL_FILE: ''
|
|
||||||
NMS_THRE: 0.6
|
|
||||||
OKS_THRE: 0.5
|
|
||||||
POST_PROCESS: false
|
|
||||||
USE_GT_BBOX: true
|
|
||||||
DEBUG:
|
|
||||||
DEBUG: true
|
|
||||||
SAVE_BATCH_IMAGES_GT: true
|
|
||||||
SAVE_BATCH_IMAGES_PRED: true
|
|
||||||
SAVE_HEATMAPS_GT: true
|
|
||||||
SAVE_HEATMAPS_PRED: true
|
|
|
@ -1,82 +0,0 @@
|
||||||
GPUS: '0,1,2,3'
|
|
||||||
BACKBONE_MODEL: pose_resnet
|
|
||||||
MODEL: multiview_pose_resnet
|
|
||||||
DATA_DIR: ''
|
|
||||||
OUTPUT_DIR: 'output'
|
|
||||||
LOG_DIR: 'log'
|
|
||||||
WORKERS: 16
|
|
||||||
PRINT_FREQ: 100
|
|
||||||
|
|
||||||
DATASET:
|
|
||||||
TRAIN_DATASET: mixed
|
|
||||||
TEST_DATASET: multiview_h36m
|
|
||||||
ROOT: '../mixed_data/'
|
|
||||||
TRAIN_SUBSET: train
|
|
||||||
TEST_SUBSET: validation
|
|
||||||
DATA_FORMAT: 'zip'
|
|
||||||
ROT_FACTOR: 0
|
|
||||||
SCALE_FACTOR: 0
|
|
||||||
NETWORK:
|
|
||||||
PRETRAINED: 'models/pytorch/imagenet/resnet152-b121ed2d.pth'
|
|
||||||
TARGET_TYPE: gaussian
|
|
||||||
AGGRE: true
|
|
||||||
HEATMAP_SIZE:
|
|
||||||
- 64
|
|
||||||
- 64
|
|
||||||
IMAGE_SIZE:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_JOINTS: 20
|
|
||||||
SIGMA: 2
|
|
||||||
POSE_RESNET:
|
|
||||||
FINAL_CONV_KERNEL: 1
|
|
||||||
DECONV_WITH_BIAS: false
|
|
||||||
NUM_DECONV_LAYERS: 3
|
|
||||||
NUM_DECONV_FILTERS:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_DECONV_KERNELS:
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
NUM_LAYERS: 152
|
|
||||||
LOSS:
|
|
||||||
USE_TARGET_WEIGHT: true
|
|
||||||
TRAIN:
|
|
||||||
BATCH_SIZE: 4
|
|
||||||
SHUFFLE: true
|
|
||||||
BEGIN_EPOCH: 0
|
|
||||||
END_EPOCH: 30
|
|
||||||
RESUME: true
|
|
||||||
OPTIMIZER: adam
|
|
||||||
LR: 0.001
|
|
||||||
LR_FACTOR: 0.1
|
|
||||||
LR_STEP:
|
|
||||||
- 20
|
|
||||||
- 25
|
|
||||||
WD: 0.0001
|
|
||||||
GAMMA1: 0.99
|
|
||||||
GAMMA2: 0.0
|
|
||||||
MOMENTUM: 0.9
|
|
||||||
NESTEROV: false
|
|
||||||
TEST:
|
|
||||||
BATCH_SIZE: 4
|
|
||||||
BBOX_FILE: ''
|
|
||||||
BBOX_THRE: 1.0
|
|
||||||
DETECTOR: fpn_dcn
|
|
||||||
DETECTOR_DIR: ''
|
|
||||||
IMAGE_THRE: 0.1
|
|
||||||
IN_VIS_THRE: 0.0
|
|
||||||
MATCH_IOU_THRE: 0.3
|
|
||||||
MODEL_FILE: ''
|
|
||||||
NMS_THRE: 0.6
|
|
||||||
OKS_THRE: 0.5
|
|
||||||
POST_PROCESS: false
|
|
||||||
USE_GT_BBOX: true
|
|
||||||
DEBUG:
|
|
||||||
DEBUG: true
|
|
||||||
SAVE_BATCH_IMAGES_GT: true
|
|
||||||
SAVE_BATCH_IMAGES_PRED: true
|
|
||||||
SAVE_HEATMAPS_GT: true
|
|
||||||
SAVE_HEATMAPS_PRED: true
|
|
|
@ -1,82 +0,0 @@
|
||||||
GPUS: '0,1,2,3'
|
|
||||||
BACKBONE_MODEL: pose_resnet
|
|
||||||
MODEL: multiview_pose_resnet
|
|
||||||
DATA_DIR: ''
|
|
||||||
OUTPUT_DIR: 'output'
|
|
||||||
LOG_DIR: 'log'
|
|
||||||
WORKERS: 16
|
|
||||||
PRINT_FREQ: 100
|
|
||||||
|
|
||||||
DATASET:
|
|
||||||
TRAIN_DATASET: mixed
|
|
||||||
TEST_DATASET: multiview_h36m
|
|
||||||
ROOT: '../mixed_data/'
|
|
||||||
TRAIN_SUBSET: train
|
|
||||||
TEST_SUBSET: validation
|
|
||||||
DATA_FORMAT: 'zip'
|
|
||||||
ROT_FACTOR: 0
|
|
||||||
SCALE_FACTOR: 0
|
|
||||||
NETWORK:
|
|
||||||
PRETRAINED: 'models/pytorch/imagenet/resnet152-b121ed2d.pth'
|
|
||||||
TARGET_TYPE: gaussian
|
|
||||||
AGGRE: false
|
|
||||||
HEATMAP_SIZE:
|
|
||||||
- 64
|
|
||||||
- 64
|
|
||||||
IMAGE_SIZE:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_JOINTS: 20
|
|
||||||
SIGMA: 2
|
|
||||||
POSE_RESNET:
|
|
||||||
FINAL_CONV_KERNEL: 1
|
|
||||||
DECONV_WITH_BIAS: false
|
|
||||||
NUM_DECONV_LAYERS: 3
|
|
||||||
NUM_DECONV_FILTERS:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_DECONV_KERNELS:
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
NUM_LAYERS: 152
|
|
||||||
LOSS:
|
|
||||||
USE_TARGET_WEIGHT: true
|
|
||||||
TRAIN:
|
|
||||||
BATCH_SIZE: 4
|
|
||||||
SHUFFLE: true
|
|
||||||
BEGIN_EPOCH: 0
|
|
||||||
END_EPOCH: 30
|
|
||||||
RESUME: true
|
|
||||||
OPTIMIZER: adam
|
|
||||||
LR: 0.001
|
|
||||||
LR_FACTOR: 0.1
|
|
||||||
LR_STEP:
|
|
||||||
- 20
|
|
||||||
- 25
|
|
||||||
WD: 0.0001
|
|
||||||
GAMMA1: 0.99
|
|
||||||
GAMMA2: 0.0
|
|
||||||
MOMENTUM: 0.9
|
|
||||||
NESTEROV: false
|
|
||||||
TEST:
|
|
||||||
BATCH_SIZE: 4
|
|
||||||
BBOX_FILE: ''
|
|
||||||
BBOX_THRE: 1.0
|
|
||||||
DETECTOR: fpn_dcn
|
|
||||||
DETECTOR_DIR: ''
|
|
||||||
IMAGE_THRE: 0.1
|
|
||||||
IN_VIS_THRE: 0.0
|
|
||||||
MATCH_IOU_THRE: 0.3
|
|
||||||
MODEL_FILE: ''
|
|
||||||
NMS_THRE: 0.6
|
|
||||||
OKS_THRE: 0.5
|
|
||||||
POST_PROCESS: false
|
|
||||||
USE_GT_BBOX: true
|
|
||||||
DEBUG:
|
|
||||||
DEBUG: true
|
|
||||||
SAVE_BATCH_IMAGES_GT: true
|
|
||||||
SAVE_BATCH_IMAGES_PRED: true
|
|
||||||
SAVE_HEATMAPS_GT: true
|
|
||||||
SAVE_HEATMAPS_PRED: true
|
|
|
@ -1,82 +0,0 @@
|
||||||
GPUS: '0,1,2,3'
|
|
||||||
BACKBONE_MODEL: pose_resnet
|
|
||||||
MODEL: multiview_pose_resnet
|
|
||||||
DATA_DIR: ''
|
|
||||||
OUTPUT_DIR: 'output'
|
|
||||||
LOG_DIR: 'log'
|
|
||||||
WORKERS: 16
|
|
||||||
PRINT_FREQ: 100
|
|
||||||
|
|
||||||
DATASET:
|
|
||||||
TRAIN_DATASET: mixed
|
|
||||||
TEST_DATASET: multiview_h36m
|
|
||||||
ROOT: '../mixed_data/'
|
|
||||||
TRAIN_SUBSET: train
|
|
||||||
TEST_SUBSET: validation
|
|
||||||
DATA_FORMAT: 'zip'
|
|
||||||
ROT_FACTOR: 0
|
|
||||||
SCALE_FACTOR: 0
|
|
||||||
NETWORK:
|
|
||||||
PRETRAINED: 'models/pytorch/imagenet/resnet152-b121ed2d.pth'
|
|
||||||
TARGET_TYPE: gaussian
|
|
||||||
AGGRE: true
|
|
||||||
HEATMAP_SIZE:
|
|
||||||
- 80
|
|
||||||
- 80
|
|
||||||
IMAGE_SIZE:
|
|
||||||
- 320
|
|
||||||
- 320
|
|
||||||
NUM_JOINTS: 20
|
|
||||||
SIGMA: 3
|
|
||||||
POSE_RESNET:
|
|
||||||
FINAL_CONV_KERNEL: 1
|
|
||||||
DECONV_WITH_BIAS: false
|
|
||||||
NUM_DECONV_LAYERS: 3
|
|
||||||
NUM_DECONV_FILTERS:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_DECONV_KERNELS:
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
NUM_LAYERS: 152
|
|
||||||
LOSS:
|
|
||||||
USE_TARGET_WEIGHT: true
|
|
||||||
TRAIN:
|
|
||||||
BATCH_SIZE: 2
|
|
||||||
SHUFFLE: true
|
|
||||||
BEGIN_EPOCH: 0
|
|
||||||
END_EPOCH: 30
|
|
||||||
RESUME: true
|
|
||||||
OPTIMIZER: adam
|
|
||||||
LR: 0.001
|
|
||||||
LR_FACTOR: 0.1
|
|
||||||
LR_STEP:
|
|
||||||
- 20
|
|
||||||
- 25
|
|
||||||
WD: 0.0001
|
|
||||||
GAMMA1: 0.99
|
|
||||||
GAMMA2: 0.0
|
|
||||||
MOMENTUM: 0.9
|
|
||||||
NESTEROV: false
|
|
||||||
TEST:
|
|
||||||
BATCH_SIZE: 2
|
|
||||||
BBOX_FILE: ''
|
|
||||||
BBOX_THRE: 1.0
|
|
||||||
DETECTOR: fpn_dcn
|
|
||||||
DETECTOR_DIR: ''
|
|
||||||
IMAGE_THRE: 0.1
|
|
||||||
IN_VIS_THRE: 0.0
|
|
||||||
MATCH_IOU_THRE: 0.3
|
|
||||||
MODEL_FILE: ''
|
|
||||||
NMS_THRE: 0.6
|
|
||||||
OKS_THRE: 0.5
|
|
||||||
POST_PROCESS: false
|
|
||||||
USE_GT_BBOX: true
|
|
||||||
DEBUG:
|
|
||||||
DEBUG: true
|
|
||||||
SAVE_BATCH_IMAGES_GT: true
|
|
||||||
SAVE_BATCH_IMAGES_PRED: true
|
|
||||||
SAVE_HEATMAPS_GT: true
|
|
||||||
SAVE_HEATMAPS_PRED: true
|
|
|
@ -1,82 +0,0 @@
|
||||||
GPUS: '0,1,2,3'
|
|
||||||
BACKBONE_MODEL: pose_resnet
|
|
||||||
MODEL: multiview_pose_resnet
|
|
||||||
DATA_DIR: ''
|
|
||||||
OUTPUT_DIR: 'output'
|
|
||||||
LOG_DIR: 'log'
|
|
||||||
WORKERS: 16
|
|
||||||
PRINT_FREQ: 100
|
|
||||||
|
|
||||||
DATASET:
|
|
||||||
TRAIN_DATASET: mixed
|
|
||||||
TEST_DATASET: multiview_h36m
|
|
||||||
ROOT: '../mixed_data/'
|
|
||||||
TRAIN_SUBSET: train
|
|
||||||
TEST_SUBSET: validation
|
|
||||||
DATA_FORMAT: 'zip'
|
|
||||||
ROT_FACTOR: 0
|
|
||||||
SCALE_FACTOR: 0
|
|
||||||
NETWORK:
|
|
||||||
PRETRAINED: 'models/pytorch/imagenet/resnet152-b121ed2d.pth'
|
|
||||||
TARGET_TYPE: gaussian
|
|
||||||
AGGRE: false
|
|
||||||
HEATMAP_SIZE:
|
|
||||||
- 80
|
|
||||||
- 80
|
|
||||||
IMAGE_SIZE:
|
|
||||||
- 320
|
|
||||||
- 320
|
|
||||||
NUM_JOINTS: 20
|
|
||||||
SIGMA: 3
|
|
||||||
POSE_RESNET:
|
|
||||||
FINAL_CONV_KERNEL: 1
|
|
||||||
DECONV_WITH_BIAS: false
|
|
||||||
NUM_DECONV_LAYERS: 3
|
|
||||||
NUM_DECONV_FILTERS:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_DECONV_KERNELS:
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
NUM_LAYERS: 152
|
|
||||||
LOSS:
|
|
||||||
USE_TARGET_WEIGHT: true
|
|
||||||
TRAIN:
|
|
||||||
BATCH_SIZE: 2
|
|
||||||
SHUFFLE: true
|
|
||||||
BEGIN_EPOCH: 0
|
|
||||||
END_EPOCH: 30
|
|
||||||
RESUME: true
|
|
||||||
OPTIMIZER: adam
|
|
||||||
LR: 0.001
|
|
||||||
LR_FACTOR: 0.1
|
|
||||||
LR_STEP:
|
|
||||||
- 20
|
|
||||||
- 25
|
|
||||||
WD: 0.0001
|
|
||||||
GAMMA1: 0.99
|
|
||||||
GAMMA2: 0.0
|
|
||||||
MOMENTUM: 0.9
|
|
||||||
NESTEROV: false
|
|
||||||
TEST:
|
|
||||||
BATCH_SIZE: 2
|
|
||||||
BBOX_FILE: ''
|
|
||||||
BBOX_THRE: 1.0
|
|
||||||
DETECTOR: fpn_dcn
|
|
||||||
DETECTOR_DIR: ''
|
|
||||||
IMAGE_THRE: 0.1
|
|
||||||
IN_VIS_THRE: 0.0
|
|
||||||
MATCH_IOU_THRE: 0.3
|
|
||||||
MODEL_FILE: ''
|
|
||||||
NMS_THRE: 0.6
|
|
||||||
OKS_THRE: 0.5
|
|
||||||
POST_PROCESS: false
|
|
||||||
USE_GT_BBOX: true
|
|
||||||
DEBUG:
|
|
||||||
DEBUG: true
|
|
||||||
SAVE_BATCH_IMAGES_GT: true
|
|
||||||
SAVE_BATCH_IMAGES_PRED: true
|
|
||||||
SAVE_HEATMAPS_GT: true
|
|
||||||
SAVE_HEATMAPS_PRED: true
|
|
|
@ -1,82 +0,0 @@
|
||||||
GPUS: '0,1,2,3'
|
|
||||||
BACKBONE_MODEL: pose_resnet
|
|
||||||
MODEL: multiview_pose_resnet
|
|
||||||
DATA_DIR: ''
|
|
||||||
OUTPUT_DIR: 'output'
|
|
||||||
LOG_DIR: 'log'
|
|
||||||
WORKERS: 16
|
|
||||||
PRINT_FREQ: 100
|
|
||||||
|
|
||||||
DATASET:
|
|
||||||
TRAIN_DATASET: mixed
|
|
||||||
TEST_DATASET: multiview_h36m
|
|
||||||
ROOT: '../mixed_data/'
|
|
||||||
TRAIN_SUBSET: train
|
|
||||||
TEST_SUBSET: validation
|
|
||||||
DATA_FORMAT: 'zip'
|
|
||||||
ROT_FACTOR: 0
|
|
||||||
SCALE_FACTOR: 0
|
|
||||||
NETWORK:
|
|
||||||
PRETRAINED: 'models/pytorch/imagenet/resnet50-19c8e357.pth'
|
|
||||||
TARGET_TYPE: gaussian
|
|
||||||
AGGRE: true
|
|
||||||
HEATMAP_SIZE:
|
|
||||||
- 64
|
|
||||||
- 64
|
|
||||||
IMAGE_SIZE:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_JOINTS: 20
|
|
||||||
SIGMA: 2
|
|
||||||
POSE_RESNET:
|
|
||||||
FINAL_CONV_KERNEL: 1
|
|
||||||
DECONV_WITH_BIAS: false
|
|
||||||
NUM_DECONV_LAYERS: 3
|
|
||||||
NUM_DECONV_FILTERS:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_DECONV_KERNELS:
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
NUM_LAYERS: 50
|
|
||||||
LOSS:
|
|
||||||
USE_TARGET_WEIGHT: true
|
|
||||||
TRAIN:
|
|
||||||
BATCH_SIZE: 4
|
|
||||||
SHUFFLE: true
|
|
||||||
BEGIN_EPOCH: 0
|
|
||||||
END_EPOCH: 30
|
|
||||||
RESUME: true
|
|
||||||
OPTIMIZER: adam
|
|
||||||
LR: 0.001
|
|
||||||
LR_FACTOR: 0.1
|
|
||||||
LR_STEP:
|
|
||||||
- 20
|
|
||||||
- 25
|
|
||||||
WD: 0.0001
|
|
||||||
GAMMA1: 0.99
|
|
||||||
GAMMA2: 0.0
|
|
||||||
MOMENTUM: 0.9
|
|
||||||
NESTEROV: false
|
|
||||||
TEST:
|
|
||||||
BATCH_SIZE: 4
|
|
||||||
BBOX_FILE: ''
|
|
||||||
BBOX_THRE: 1.0
|
|
||||||
DETECTOR: fpn_dcn
|
|
||||||
DETECTOR_DIR: ''
|
|
||||||
IMAGE_THRE: 0.1
|
|
||||||
IN_VIS_THRE: 0.0
|
|
||||||
MATCH_IOU_THRE: 0.3
|
|
||||||
MODEL_FILE: ''
|
|
||||||
NMS_THRE: 0.6
|
|
||||||
OKS_THRE: 0.5
|
|
||||||
POST_PROCESS: false
|
|
||||||
USE_GT_BBOX: true
|
|
||||||
DEBUG:
|
|
||||||
DEBUG: true
|
|
||||||
SAVE_BATCH_IMAGES_GT: true
|
|
||||||
SAVE_BATCH_IMAGES_PRED: true
|
|
||||||
SAVE_HEATMAPS_GT: true
|
|
||||||
SAVE_HEATMAPS_PRED: true
|
|
|
@ -1,82 +0,0 @@
|
||||||
GPUS: '0,1,2,3'
|
|
||||||
BACKBONE_MODEL: pose_resnet
|
|
||||||
MODEL: multiview_pose_resnet
|
|
||||||
DATA_DIR: ''
|
|
||||||
OUTPUT_DIR: 'output'
|
|
||||||
LOG_DIR: 'log'
|
|
||||||
WORKERS: 16
|
|
||||||
PRINT_FREQ: 100
|
|
||||||
|
|
||||||
DATASET:
|
|
||||||
TRAIN_DATASET: mixed
|
|
||||||
TEST_DATASET: multiview_h36m
|
|
||||||
ROOT: '../mixed_data/'
|
|
||||||
TRAIN_SUBSET: train
|
|
||||||
TEST_SUBSET: validation
|
|
||||||
DATA_FORMAT: 'zip'
|
|
||||||
ROT_FACTOR: 0
|
|
||||||
SCALE_FACTOR: 0
|
|
||||||
NETWORK:
|
|
||||||
PRETRAINED: 'models/pytorch/imagenet/resnet50-19c8e357.pth'
|
|
||||||
TARGET_TYPE: gaussian
|
|
||||||
AGGRE: false
|
|
||||||
HEATMAP_SIZE:
|
|
||||||
- 64
|
|
||||||
- 64
|
|
||||||
IMAGE_SIZE:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_JOINTS: 20
|
|
||||||
SIGMA: 2
|
|
||||||
POSE_RESNET:
|
|
||||||
FINAL_CONV_KERNEL: 1
|
|
||||||
DECONV_WITH_BIAS: false
|
|
||||||
NUM_DECONV_LAYERS: 3
|
|
||||||
NUM_DECONV_FILTERS:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_DECONV_KERNELS:
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
NUM_LAYERS: 50
|
|
||||||
LOSS:
|
|
||||||
USE_TARGET_WEIGHT: true
|
|
||||||
TRAIN:
|
|
||||||
BATCH_SIZE: 4
|
|
||||||
SHUFFLE: true
|
|
||||||
BEGIN_EPOCH: 0
|
|
||||||
END_EPOCH: 30
|
|
||||||
RESUME: true
|
|
||||||
OPTIMIZER: adam
|
|
||||||
LR: 0.001
|
|
||||||
LR_FACTOR: 0.1
|
|
||||||
LR_STEP:
|
|
||||||
- 20
|
|
||||||
- 25
|
|
||||||
WD: 0.0001
|
|
||||||
GAMMA1: 0.99
|
|
||||||
GAMMA2: 0.0
|
|
||||||
MOMENTUM: 0.9
|
|
||||||
NESTEROV: false
|
|
||||||
TEST:
|
|
||||||
BATCH_SIZE: 4
|
|
||||||
BBOX_FILE: ''
|
|
||||||
BBOX_THRE: 1.0
|
|
||||||
DETECTOR: fpn_dcn
|
|
||||||
DETECTOR_DIR: ''
|
|
||||||
IMAGE_THRE: 0.1
|
|
||||||
IN_VIS_THRE: 0.0
|
|
||||||
MATCH_IOU_THRE: 0.3
|
|
||||||
MODEL_FILE: ''
|
|
||||||
NMS_THRE: 0.6
|
|
||||||
OKS_THRE: 0.5
|
|
||||||
POST_PROCESS: false
|
|
||||||
USE_GT_BBOX: true
|
|
||||||
DEBUG:
|
|
||||||
DEBUG: true
|
|
||||||
SAVE_BATCH_IMAGES_GT: true
|
|
||||||
SAVE_BATCH_IMAGES_PRED: true
|
|
||||||
SAVE_HEATMAPS_GT: true
|
|
||||||
SAVE_HEATMAPS_PRED: true
|
|
|
@ -1,82 +0,0 @@
|
||||||
GPUS: '0,1,2,3'
|
|
||||||
BACKBONE_MODEL: pose_resnet
|
|
||||||
MODEL: multiview_pose_resnet
|
|
||||||
DATA_DIR: ''
|
|
||||||
OUTPUT_DIR: 'output'
|
|
||||||
LOG_DIR: 'log'
|
|
||||||
WORKERS: 16
|
|
||||||
PRINT_FREQ: 100
|
|
||||||
|
|
||||||
DATASET:
|
|
||||||
TRAIN_DATASET: mixed
|
|
||||||
TEST_DATASET: multiview_h36m
|
|
||||||
ROOT: '../mixed_data/'
|
|
||||||
TRAIN_SUBSET: train
|
|
||||||
TEST_SUBSET: validation
|
|
||||||
DATA_FORMAT: 'zip'
|
|
||||||
ROT_FACTOR: 0
|
|
||||||
SCALE_FACTOR: 0
|
|
||||||
NETWORK:
|
|
||||||
PRETRAINED: 'models/pytorch/imagenet/resnet50-19c8e357.pth'
|
|
||||||
TARGET_TYPE: gaussian
|
|
||||||
AGGRE: true
|
|
||||||
HEATMAP_SIZE:
|
|
||||||
- 80
|
|
||||||
- 80
|
|
||||||
IMAGE_SIZE:
|
|
||||||
- 320
|
|
||||||
- 320
|
|
||||||
NUM_JOINTS: 20
|
|
||||||
SIGMA: 3
|
|
||||||
POSE_RESNET:
|
|
||||||
FINAL_CONV_KERNEL: 1
|
|
||||||
DECONV_WITH_BIAS: false
|
|
||||||
NUM_DECONV_LAYERS: 3
|
|
||||||
NUM_DECONV_FILTERS:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_DECONV_KERNELS:
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
NUM_LAYERS: 50
|
|
||||||
LOSS:
|
|
||||||
USE_TARGET_WEIGHT: true
|
|
||||||
TRAIN:
|
|
||||||
BATCH_SIZE: 2
|
|
||||||
SHUFFLE: true
|
|
||||||
BEGIN_EPOCH: 0
|
|
||||||
END_EPOCH: 30
|
|
||||||
RESUME: true
|
|
||||||
OPTIMIZER: adam
|
|
||||||
LR: 0.001
|
|
||||||
LR_FACTOR: 0.1
|
|
||||||
LR_STEP:
|
|
||||||
- 20
|
|
||||||
- 25
|
|
||||||
WD: 0.0001
|
|
||||||
GAMMA1: 0.99
|
|
||||||
GAMMA2: 0.0
|
|
||||||
MOMENTUM: 0.9
|
|
||||||
NESTEROV: false
|
|
||||||
TEST:
|
|
||||||
BATCH_SIZE: 2
|
|
||||||
BBOX_FILE: ''
|
|
||||||
BBOX_THRE: 1.0
|
|
||||||
DETECTOR: fpn_dcn
|
|
||||||
DETECTOR_DIR: ''
|
|
||||||
IMAGE_THRE: 0.1
|
|
||||||
IN_VIS_THRE: 0.0
|
|
||||||
MATCH_IOU_THRE: 0.3
|
|
||||||
MODEL_FILE: ''
|
|
||||||
NMS_THRE: 0.6
|
|
||||||
OKS_THRE: 0.5
|
|
||||||
POST_PROCESS: false
|
|
||||||
USE_GT_BBOX: true
|
|
||||||
DEBUG:
|
|
||||||
DEBUG: true
|
|
||||||
SAVE_BATCH_IMAGES_GT: true
|
|
||||||
SAVE_BATCH_IMAGES_PRED: true
|
|
||||||
SAVE_HEATMAPS_GT: true
|
|
||||||
SAVE_HEATMAPS_PRED: true
|
|
|
@ -1,82 +0,0 @@
|
||||||
GPUS: '0,1,2,3'
|
|
||||||
BACKBONE_MODEL: pose_resnet
|
|
||||||
MODEL: multiview_pose_resnet
|
|
||||||
DATA_DIR: ''
|
|
||||||
OUTPUT_DIR: 'output'
|
|
||||||
LOG_DIR: 'log'
|
|
||||||
WORKERS: 16
|
|
||||||
PRINT_FREQ: 100
|
|
||||||
|
|
||||||
DATASET:
|
|
||||||
TRAIN_DATASET: mixed
|
|
||||||
TEST_DATASET: multiview_h36m
|
|
||||||
ROOT: '../mixed_data/'
|
|
||||||
TRAIN_SUBSET: train
|
|
||||||
TEST_SUBSET: validation
|
|
||||||
DATA_FORMAT: 'zip'
|
|
||||||
ROT_FACTOR: 0
|
|
||||||
SCALE_FACTOR: 0
|
|
||||||
NETWORK:
|
|
||||||
PRETRAINED: 'models/pytorch/imagenet/resnet50-19c8e357.pth'
|
|
||||||
TARGET_TYPE: gaussian
|
|
||||||
AGGRE: false
|
|
||||||
HEATMAP_SIZE:
|
|
||||||
- 80
|
|
||||||
- 80
|
|
||||||
IMAGE_SIZE:
|
|
||||||
- 320
|
|
||||||
- 320
|
|
||||||
NUM_JOINTS: 20
|
|
||||||
SIGMA: 3
|
|
||||||
POSE_RESNET:
|
|
||||||
FINAL_CONV_KERNEL: 1
|
|
||||||
DECONV_WITH_BIAS: false
|
|
||||||
NUM_DECONV_LAYERS: 3
|
|
||||||
NUM_DECONV_FILTERS:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_DECONV_KERNELS:
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
NUM_LAYERS: 50
|
|
||||||
LOSS:
|
|
||||||
USE_TARGET_WEIGHT: true
|
|
||||||
TRAIN:
|
|
||||||
BATCH_SIZE: 2
|
|
||||||
SHUFFLE: true
|
|
||||||
BEGIN_EPOCH: 0
|
|
||||||
END_EPOCH: 30
|
|
||||||
RESUME: true
|
|
||||||
OPTIMIZER: adam
|
|
||||||
LR: 0.001
|
|
||||||
LR_FACTOR: 0.1
|
|
||||||
LR_STEP:
|
|
||||||
- 20
|
|
||||||
- 25
|
|
||||||
WD: 0.0001
|
|
||||||
GAMMA1: 0.99
|
|
||||||
GAMMA2: 0.0
|
|
||||||
MOMENTUM: 0.9
|
|
||||||
NESTEROV: false
|
|
||||||
TEST:
|
|
||||||
BATCH_SIZE: 2
|
|
||||||
BBOX_FILE: ''
|
|
||||||
BBOX_THRE: 1.0
|
|
||||||
DETECTOR: fpn_dcn
|
|
||||||
DETECTOR_DIR: ''
|
|
||||||
IMAGE_THRE: 0.1
|
|
||||||
IN_VIS_THRE: 0.0
|
|
||||||
MATCH_IOU_THRE: 0.3
|
|
||||||
MODEL_FILE: ''
|
|
||||||
NMS_THRE: 0.6
|
|
||||||
OKS_THRE: 0.5
|
|
||||||
POST_PROCESS: false
|
|
||||||
USE_GT_BBOX: true
|
|
||||||
DEBUG:
|
|
||||||
DEBUG: true
|
|
||||||
SAVE_BATCH_IMAGES_GT: true
|
|
||||||
SAVE_BATCH_IMAGES_PRED: true
|
|
||||||
SAVE_HEATMAPS_GT: true
|
|
||||||
SAVE_HEATMAPS_PRED: true
|
|
|
@ -1,84 +0,0 @@
|
||||||
GPUS: '0'
|
|
||||||
BACKBONE_MODEL: pose_resnet
|
|
||||||
MODEL: multiview_pose_resnet
|
|
||||||
DATA_DIR: ''
|
|
||||||
OUTPUT_DIR: 'output'
|
|
||||||
LOG_DIR: 'log'
|
|
||||||
WORKERS: 8
|
|
||||||
PRINT_FREQ: 100
|
|
||||||
|
|
||||||
DATASET:
|
|
||||||
TRAIN_DATASET: mixed
|
|
||||||
TRAIN_SUBSET: train
|
|
||||||
TEST_DATASET: multiview_h36m
|
|
||||||
TEST_SUBSET: validation
|
|
||||||
ROOT: 'data/'
|
|
||||||
DATA_FORMAT: 'jpg'
|
|
||||||
ROT_FACTOR: 0
|
|
||||||
SCALE_FACTOR: 0
|
|
||||||
ROOTIDX: 0
|
|
||||||
NETWORK:
|
|
||||||
PRETRAINED: 'models/pytorch/imagenet/resnet152-b121ed2d.pth'
|
|
||||||
TARGET_TYPE: gaussian
|
|
||||||
AGGRE: false
|
|
||||||
HEATMAP_SIZE:
|
|
||||||
- 64
|
|
||||||
- 64
|
|
||||||
IMAGE_SIZE:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_JOINTS: 20
|
|
||||||
SIGMA: 2
|
|
||||||
POSE_RESNET:
|
|
||||||
FINAL_CONV_KERNEL: 1
|
|
||||||
DECONV_WITH_BIAS: false
|
|
||||||
NUM_DECONV_LAYERS: 3
|
|
||||||
NUM_DECONV_FILTERS:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_DECONV_KERNELS:
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
NUM_LAYERS: 152
|
|
||||||
LOSS:
|
|
||||||
USE_TARGET_WEIGHT: true
|
|
||||||
TRAIN:
|
|
||||||
BATCH_SIZE: 8
|
|
||||||
SHUFFLE: true
|
|
||||||
MODEL_FILE: 'models/pytorch/multiview_pose_resnet/multiview_pose_resnet152_mixed_noaggre_best.pth.tar'
|
|
||||||
BEGIN_EPOCH: 0
|
|
||||||
END_EPOCH: 30
|
|
||||||
RESUME: false
|
|
||||||
OPTIMIZER: adam
|
|
||||||
LR: 0.001
|
|
||||||
LR_FACTOR: 0.1
|
|
||||||
LR_STEP:
|
|
||||||
- 20
|
|
||||||
- 25
|
|
||||||
WD: 0.0001
|
|
||||||
GAMMA1: 0.99
|
|
||||||
GAMMA2: 0.0
|
|
||||||
MOMENTUM: 0.9
|
|
||||||
NESTEROV: false
|
|
||||||
TEST:
|
|
||||||
BATCH_SIZE: 8
|
|
||||||
BBOX_FILE: ''
|
|
||||||
BBOX_THRE: 1.0
|
|
||||||
DETECTOR: fpn_dcn
|
|
||||||
DETECTOR_DIR: ''
|
|
||||||
IMAGE_THRE: 0.1
|
|
||||||
IN_VIS_THRE: 0.0
|
|
||||||
MATCH_IOU_THRE: 0.3
|
|
||||||
MODEL_FILE: ''
|
|
||||||
NMS_THRE: 0.6
|
|
||||||
OKS_THRE: 0.5
|
|
||||||
POST_PROCESS: false
|
|
||||||
USE_GT_BBOX: true
|
|
||||||
DEBUG:
|
|
||||||
DEBUG: true
|
|
||||||
SAVE_BATCH_IMAGES_GT: true
|
|
||||||
SAVE_BATCH_IMAGES_PRED: true
|
|
||||||
SAVE_HEATMAPS_GT: true
|
|
||||||
SAVE_HEATMAPS_PRED: true
|
|
|
@ -1,82 +0,0 @@
|
||||||
GPUS: '0,1,2,3'
|
|
||||||
BACKBONE_MODEL: pose_mobilenetv2
|
|
||||||
MODEL: multiview_pose_mobilenetv2
|
|
||||||
DATA_DIR: ''
|
|
||||||
OUTPUT_DIR: 'output'
|
|
||||||
LOG_DIR: 'log'
|
|
||||||
WORKERS: 16
|
|
||||||
PRINT_FREQ: 100
|
|
||||||
|
|
||||||
DATASET:
|
|
||||||
TRAIN_DATASET: multiview_mpii
|
|
||||||
TEST_DATASET: multiview_h36m
|
|
||||||
ROOT: '../mixed_data/'
|
|
||||||
TRAIN_SUBSET: train
|
|
||||||
TEST_SUBSET: validation
|
|
||||||
DATA_FORMAT: 'zip'
|
|
||||||
ROT_FACTOR: 0
|
|
||||||
SCALE_FACTOR: 0
|
|
||||||
NETWORK:
|
|
||||||
PRETRAINED: 'models/pytorch/imagenet/mobilenet_v2.pth.tar'
|
|
||||||
TARGET_TYPE: gaussian
|
|
||||||
AGGRE: false
|
|
||||||
HEATMAP_SIZE:
|
|
||||||
- 64
|
|
||||||
- 64
|
|
||||||
IMAGE_SIZE:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_JOINTS: 20
|
|
||||||
SIGMA: 2
|
|
||||||
POSE_RESNET:
|
|
||||||
FINAL_CONV_KERNEL: 1
|
|
||||||
DECONV_WITH_BIAS: false
|
|
||||||
NUM_DECONV_LAYERS: 3
|
|
||||||
NUM_DECONV_FILTERS:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_DECONV_KERNELS:
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
NUM_LAYERS: 50
|
|
||||||
LOSS:
|
|
||||||
USE_TARGET_WEIGHT: true
|
|
||||||
TRAIN:
|
|
||||||
BATCH_SIZE: 8
|
|
||||||
SHUFFLE: true
|
|
||||||
BEGIN_EPOCH: 0
|
|
||||||
END_EPOCH: 140
|
|
||||||
RESUME: true
|
|
||||||
OPTIMIZER: adam
|
|
||||||
LR: 0.001
|
|
||||||
LR_FACTOR: 0.1
|
|
||||||
LR_STEP:
|
|
||||||
- 90
|
|
||||||
- 120
|
|
||||||
WD: 0.0001
|
|
||||||
GAMMA1: 0.99
|
|
||||||
GAMMA2: 0.0
|
|
||||||
MOMENTUM: 0.9
|
|
||||||
NESTEROV: false
|
|
||||||
TEST:
|
|
||||||
BATCH_SIZE: 4
|
|
||||||
BBOX_FILE: ''
|
|
||||||
BBOX_THRE: 1.0
|
|
||||||
DETECTOR: fpn_dcn
|
|
||||||
DETECTOR_DIR: ''
|
|
||||||
IMAGE_THRE: 0.1
|
|
||||||
IN_VIS_THRE: 0.0
|
|
||||||
MATCH_IOU_THRE: 0.3
|
|
||||||
MODEL_FILE: ''
|
|
||||||
NMS_THRE: 0.6
|
|
||||||
OKS_THRE: 0.5
|
|
||||||
POST_PROCESS: false
|
|
||||||
USE_GT_BBOX: true
|
|
||||||
DEBUG:
|
|
||||||
DEBUG: true
|
|
||||||
SAVE_BATCH_IMAGES_GT: true
|
|
||||||
SAVE_BATCH_IMAGES_PRED: true
|
|
||||||
SAVE_HEATMAPS_GT: true
|
|
||||||
SAVE_HEATMAPS_PRED: true
|
|
|
@ -1,82 +0,0 @@
|
||||||
GPUS: '0,1,2,3'
|
|
||||||
BACKBONE_MODEL: pose_mobilenetv2
|
|
||||||
MODEL: multiview_pose_mobilenetv2
|
|
||||||
DATA_DIR: ''
|
|
||||||
OUTPUT_DIR: 'output'
|
|
||||||
LOG_DIR: 'log'
|
|
||||||
WORKERS: 16
|
|
||||||
PRINT_FREQ: 100
|
|
||||||
|
|
||||||
DATASET:
|
|
||||||
TRAIN_DATASET: multiview_mpii
|
|
||||||
TEST_DATASET: multiview_h36m
|
|
||||||
ROOT: '../mixed_data/'
|
|
||||||
TRAIN_SUBSET: train
|
|
||||||
TEST_SUBSET: validation
|
|
||||||
DATA_FORMAT: 'zip'
|
|
||||||
ROT_FACTOR: 0
|
|
||||||
SCALE_FACTOR: 0
|
|
||||||
NETWORK:
|
|
||||||
PRETRAINED: 'models/pytorch/imagenet/mobilenet_v2.pth.tar'
|
|
||||||
TARGET_TYPE: gaussian
|
|
||||||
AGGRE: false
|
|
||||||
HEATMAP_SIZE:
|
|
||||||
- 80
|
|
||||||
- 80
|
|
||||||
IMAGE_SIZE:
|
|
||||||
- 320
|
|
||||||
- 320
|
|
||||||
NUM_JOINTS: 20
|
|
||||||
SIGMA: 3
|
|
||||||
POSE_RESNET:
|
|
||||||
FINAL_CONV_KERNEL: 1
|
|
||||||
DECONV_WITH_BIAS: false
|
|
||||||
NUM_DECONV_LAYERS: 3
|
|
||||||
NUM_DECONV_FILTERS:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_DECONV_KERNELS:
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
NUM_LAYERS: 50
|
|
||||||
LOSS:
|
|
||||||
USE_TARGET_WEIGHT: true
|
|
||||||
TRAIN:
|
|
||||||
BATCH_SIZE: 4
|
|
||||||
SHUFFLE: true
|
|
||||||
BEGIN_EPOCH: 0
|
|
||||||
END_EPOCH: 140
|
|
||||||
RESUME: true
|
|
||||||
OPTIMIZER: adam
|
|
||||||
LR: 0.001
|
|
||||||
LR_FACTOR: 0.1
|
|
||||||
LR_STEP:
|
|
||||||
- 90
|
|
||||||
- 120
|
|
||||||
WD: 0.0001
|
|
||||||
GAMMA1: 0.99
|
|
||||||
GAMMA2: 0.0
|
|
||||||
MOMENTUM: 0.9
|
|
||||||
NESTEROV: false
|
|
||||||
TEST:
|
|
||||||
BATCH_SIZE: 4
|
|
||||||
BBOX_FILE: ''
|
|
||||||
BBOX_THRE: 1.0
|
|
||||||
DETECTOR: fpn_dcn
|
|
||||||
DETECTOR_DIR: ''
|
|
||||||
IMAGE_THRE: 0.1
|
|
||||||
IN_VIS_THRE: 0.0
|
|
||||||
MATCH_IOU_THRE: 0.3
|
|
||||||
MODEL_FILE: ''
|
|
||||||
NMS_THRE: 0.6
|
|
||||||
OKS_THRE: 0.5
|
|
||||||
POST_PROCESS: false
|
|
||||||
USE_GT_BBOX: true
|
|
||||||
DEBUG:
|
|
||||||
DEBUG: true
|
|
||||||
SAVE_BATCH_IMAGES_GT: true
|
|
||||||
SAVE_BATCH_IMAGES_PRED: true
|
|
||||||
SAVE_HEATMAPS_GT: true
|
|
||||||
SAVE_HEATMAPS_PRED: true
|
|
|
@ -1,82 +0,0 @@
|
||||||
GPUS: '0,1,2,3'
|
|
||||||
BACKBONE_MODEL: pose_resnet
|
|
||||||
MODEL: multiview_pose_resnet
|
|
||||||
DATA_DIR: ''
|
|
||||||
OUTPUT_DIR: 'output'
|
|
||||||
LOG_DIR: 'log'
|
|
||||||
WORKERS: 16
|
|
||||||
PRINT_FREQ: 100
|
|
||||||
|
|
||||||
DATASET:
|
|
||||||
TRAIN_DATASET: multiview_mpii
|
|
||||||
TEST_DATASET: multiview_h36m
|
|
||||||
ROOT: '../mixed_data/'
|
|
||||||
TRAIN_SUBSET: train
|
|
||||||
TEST_SUBSET: validation
|
|
||||||
DATA_FORMAT: 'zip'
|
|
||||||
ROT_FACTOR: 0
|
|
||||||
SCALE_FACTOR: 0
|
|
||||||
NETWORK:
|
|
||||||
PRETRAINED: 'models/pytorch/imagenet/resnet152-b121ed2d.pth'
|
|
||||||
TARGET_TYPE: gaussian
|
|
||||||
AGGRE: false
|
|
||||||
HEATMAP_SIZE:
|
|
||||||
- 64
|
|
||||||
- 64
|
|
||||||
IMAGE_SIZE:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_JOINTS: 20
|
|
||||||
SIGMA: 2
|
|
||||||
POSE_RESNET:
|
|
||||||
FINAL_CONV_KERNEL: 1
|
|
||||||
DECONV_WITH_BIAS: false
|
|
||||||
NUM_DECONV_LAYERS: 3
|
|
||||||
NUM_DECONV_FILTERS:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_DECONV_KERNELS:
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
NUM_LAYERS: 152
|
|
||||||
LOSS:
|
|
||||||
USE_TARGET_WEIGHT: true
|
|
||||||
TRAIN:
|
|
||||||
BATCH_SIZE: 4
|
|
||||||
SHUFFLE: true
|
|
||||||
BEGIN_EPOCH: 0
|
|
||||||
END_EPOCH: 140
|
|
||||||
RESUME: true
|
|
||||||
OPTIMIZER: adam
|
|
||||||
LR: 0.001
|
|
||||||
LR_FACTOR: 0.1
|
|
||||||
LR_STEP:
|
|
||||||
- 90
|
|
||||||
- 120
|
|
||||||
WD: 0.0001
|
|
||||||
GAMMA1: 0.99
|
|
||||||
GAMMA2: 0.0
|
|
||||||
MOMENTUM: 0.9
|
|
||||||
NESTEROV: false
|
|
||||||
TEST:
|
|
||||||
BATCH_SIZE: 4
|
|
||||||
BBOX_FILE: ''
|
|
||||||
BBOX_THRE: 1.0
|
|
||||||
DETECTOR: fpn_dcn
|
|
||||||
DETECTOR_DIR: ''
|
|
||||||
IMAGE_THRE: 0.1
|
|
||||||
IN_VIS_THRE: 0.0
|
|
||||||
MATCH_IOU_THRE: 0.3
|
|
||||||
MODEL_FILE: ''
|
|
||||||
NMS_THRE: 0.6
|
|
||||||
OKS_THRE: 0.5
|
|
||||||
POST_PROCESS: false
|
|
||||||
USE_GT_BBOX: true
|
|
||||||
DEBUG:
|
|
||||||
DEBUG: true
|
|
||||||
SAVE_BATCH_IMAGES_GT: true
|
|
||||||
SAVE_BATCH_IMAGES_PRED: true
|
|
||||||
SAVE_HEATMAPS_GT: true
|
|
||||||
SAVE_HEATMAPS_PRED: true
|
|
|
@ -1,82 +0,0 @@
|
||||||
GPUS: '0,1,2,3'
|
|
||||||
BACKBONE_MODEL: pose_resnet
|
|
||||||
MODEL: multiview_pose_resnet
|
|
||||||
DATA_DIR: ''
|
|
||||||
OUTPUT_DIR: 'output'
|
|
||||||
LOG_DIR: 'log'
|
|
||||||
WORKERS: 16
|
|
||||||
PRINT_FREQ: 100
|
|
||||||
|
|
||||||
DATASET:
|
|
||||||
TRAIN_DATASET: multiview_mpii
|
|
||||||
TEST_DATASET: multiview_h36m
|
|
||||||
ROOT: '../mixed_data/'
|
|
||||||
TRAIN_SUBSET: train
|
|
||||||
TEST_SUBSET: validation
|
|
||||||
DATA_FORMAT: 'zip'
|
|
||||||
ROT_FACTOR: 0
|
|
||||||
SCALE_FACTOR: 0
|
|
||||||
NETWORK:
|
|
||||||
PRETRAINED: 'models/pytorch/imagenet/resnet152-b121ed2d.pth'
|
|
||||||
TARGET_TYPE: gaussian
|
|
||||||
AGGRE: false
|
|
||||||
HEATMAP_SIZE:
|
|
||||||
- 80
|
|
||||||
- 80
|
|
||||||
IMAGE_SIZE:
|
|
||||||
- 320
|
|
||||||
- 320
|
|
||||||
NUM_JOINTS: 20
|
|
||||||
SIGMA: 3
|
|
||||||
POSE_RESNET:
|
|
||||||
FINAL_CONV_KERNEL: 1
|
|
||||||
DECONV_WITH_BIAS: false
|
|
||||||
NUM_DECONV_LAYERS: 3
|
|
||||||
NUM_DECONV_FILTERS:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_DECONV_KERNELS:
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
NUM_LAYERS: 152
|
|
||||||
LOSS:
|
|
||||||
USE_TARGET_WEIGHT: true
|
|
||||||
TRAIN:
|
|
||||||
BATCH_SIZE: 2
|
|
||||||
SHUFFLE: true
|
|
||||||
BEGIN_EPOCH: 0
|
|
||||||
END_EPOCH: 140
|
|
||||||
RESUME: true
|
|
||||||
OPTIMIZER: adam
|
|
||||||
LR: 0.001
|
|
||||||
LR_FACTOR: 0.1
|
|
||||||
LR_STEP:
|
|
||||||
- 90
|
|
||||||
- 120
|
|
||||||
WD: 0.0001
|
|
||||||
GAMMA1: 0.99
|
|
||||||
GAMMA2: 0.0
|
|
||||||
MOMENTUM: 0.9
|
|
||||||
NESTEROV: false
|
|
||||||
TEST:
|
|
||||||
BATCH_SIZE: 2
|
|
||||||
BBOX_FILE: ''
|
|
||||||
BBOX_THRE: 1.0
|
|
||||||
DETECTOR: fpn_dcn
|
|
||||||
DETECTOR_DIR: ''
|
|
||||||
IMAGE_THRE: 0.1
|
|
||||||
IN_VIS_THRE: 0.0
|
|
||||||
MATCH_IOU_THRE: 0.3
|
|
||||||
MODEL_FILE: ''
|
|
||||||
NMS_THRE: 0.6
|
|
||||||
OKS_THRE: 0.5
|
|
||||||
POST_PROCESS: false
|
|
||||||
USE_GT_BBOX: true
|
|
||||||
DEBUG:
|
|
||||||
DEBUG: true
|
|
||||||
SAVE_BATCH_IMAGES_GT: true
|
|
||||||
SAVE_BATCH_IMAGES_PRED: true
|
|
||||||
SAVE_HEATMAPS_GT: true
|
|
||||||
SAVE_HEATMAPS_PRED: true
|
|
|
@ -1,82 +0,0 @@
|
||||||
GPUS: '0,1,2,3'
|
|
||||||
BACKBONE_MODEL: pose_resnet
|
|
||||||
MODEL: multiview_pose_resnet
|
|
||||||
DATA_DIR: ''
|
|
||||||
OUTPUT_DIR: 'output'
|
|
||||||
LOG_DIR: 'log'
|
|
||||||
WORKERS: 16
|
|
||||||
PRINT_FREQ: 100
|
|
||||||
|
|
||||||
DATASET:
|
|
||||||
TRAIN_DATASET: multiview_mpii
|
|
||||||
TEST_DATASET: multiview_h36m
|
|
||||||
ROOT: '../mixed_data/'
|
|
||||||
TRAIN_SUBSET: train
|
|
||||||
TEST_SUBSET: validation
|
|
||||||
DATA_FORMAT: 'zip'
|
|
||||||
ROT_FACTOR: 0
|
|
||||||
SCALE_FACTOR: 0
|
|
||||||
NETWORK:
|
|
||||||
PRETRAINED: 'models/pytorch/imagenet/resnet50-19c8e357.pth'
|
|
||||||
TARGET_TYPE: gaussian
|
|
||||||
AGGRE: false
|
|
||||||
HEATMAP_SIZE:
|
|
||||||
- 64
|
|
||||||
- 64
|
|
||||||
IMAGE_SIZE:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_JOINTS: 20
|
|
||||||
SIGMA: 2
|
|
||||||
POSE_RESNET:
|
|
||||||
FINAL_CONV_KERNEL: 1
|
|
||||||
DECONV_WITH_BIAS: false
|
|
||||||
NUM_DECONV_LAYERS: 3
|
|
||||||
NUM_DECONV_FILTERS:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_DECONV_KERNELS:
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
NUM_LAYERS: 50
|
|
||||||
LOSS:
|
|
||||||
USE_TARGET_WEIGHT: true
|
|
||||||
TRAIN:
|
|
||||||
BATCH_SIZE: 4
|
|
||||||
SHUFFLE: true
|
|
||||||
BEGIN_EPOCH: 0
|
|
||||||
END_EPOCH: 140
|
|
||||||
RESUME: true
|
|
||||||
OPTIMIZER: adam
|
|
||||||
LR: 0.001
|
|
||||||
LR_FACTOR: 0.1
|
|
||||||
LR_STEP:
|
|
||||||
- 90
|
|
||||||
- 120
|
|
||||||
WD: 0.0001
|
|
||||||
GAMMA1: 0.99
|
|
||||||
GAMMA2: 0.0
|
|
||||||
MOMENTUM: 0.9
|
|
||||||
NESTEROV: false
|
|
||||||
TEST:
|
|
||||||
BATCH_SIZE: 4
|
|
||||||
BBOX_FILE: ''
|
|
||||||
BBOX_THRE: 1.0
|
|
||||||
DETECTOR: fpn_dcn
|
|
||||||
DETECTOR_DIR: ''
|
|
||||||
IMAGE_THRE: 0.1
|
|
||||||
IN_VIS_THRE: 0.0
|
|
||||||
MATCH_IOU_THRE: 0.3
|
|
||||||
MODEL_FILE: ''
|
|
||||||
NMS_THRE: 0.6
|
|
||||||
OKS_THRE: 0.5
|
|
||||||
POST_PROCESS: false
|
|
||||||
USE_GT_BBOX: true
|
|
||||||
DEBUG:
|
|
||||||
DEBUG: true
|
|
||||||
SAVE_BATCH_IMAGES_GT: true
|
|
||||||
SAVE_BATCH_IMAGES_PRED: true
|
|
||||||
SAVE_HEATMAPS_GT: true
|
|
||||||
SAVE_HEATMAPS_PRED: true
|
|
|
@ -1,82 +0,0 @@
|
||||||
GPUS: '0,1,2,3'
|
|
||||||
BACKBONE_MODEL: pose_resnet
|
|
||||||
MODEL: multiview_pose_resnet
|
|
||||||
DATA_DIR: ''
|
|
||||||
OUTPUT_DIR: 'output'
|
|
||||||
LOG_DIR: 'log'
|
|
||||||
WORKERS: 16
|
|
||||||
PRINT_FREQ: 100
|
|
||||||
|
|
||||||
DATASET:
|
|
||||||
TRAIN_DATASET: multiview_mpii
|
|
||||||
TEST_DATASET: multiview_h36m
|
|
||||||
ROOT: '../mixed_data/'
|
|
||||||
TRAIN_SUBSET: train
|
|
||||||
TEST_SUBSET: validation
|
|
||||||
DATA_FORMAT: 'zip'
|
|
||||||
ROT_FACTOR: 0
|
|
||||||
SCALE_FACTOR: 0
|
|
||||||
NETWORK:
|
|
||||||
PRETRAINED: 'models/pytorch/imagenet/resnet50-19c8e357.pth'
|
|
||||||
TARGET_TYPE: gaussian
|
|
||||||
AGGRE: false
|
|
||||||
HEATMAP_SIZE:
|
|
||||||
- 80
|
|
||||||
- 80
|
|
||||||
IMAGE_SIZE:
|
|
||||||
- 320
|
|
||||||
- 320
|
|
||||||
NUM_JOINTS: 20
|
|
||||||
SIGMA: 3
|
|
||||||
POSE_RESNET:
|
|
||||||
FINAL_CONV_KERNEL: 1
|
|
||||||
DECONV_WITH_BIAS: false
|
|
||||||
NUM_DECONV_LAYERS: 3
|
|
||||||
NUM_DECONV_FILTERS:
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
- 256
|
|
||||||
NUM_DECONV_KERNELS:
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
- 4
|
|
||||||
NUM_LAYERS: 50
|
|
||||||
LOSS:
|
|
||||||
USE_TARGET_WEIGHT: true
|
|
||||||
TRAIN:
|
|
||||||
BATCH_SIZE: 2
|
|
||||||
SHUFFLE: true
|
|
||||||
BEGIN_EPOCH: 0
|
|
||||||
END_EPOCH: 140
|
|
||||||
RESUME: true
|
|
||||||
OPTIMIZER: adam
|
|
||||||
LR: 0.001
|
|
||||||
LR_FACTOR: 0.1
|
|
||||||
LR_STEP:
|
|
||||||
- 90
|
|
||||||
- 120
|
|
||||||
WD: 0.0001
|
|
||||||
GAMMA1: 0.99
|
|
||||||
GAMMA2: 0.0
|
|
||||||
MOMENTUM: 0.9
|
|
||||||
NESTEROV: false
|
|
||||||
TEST:
|
|
||||||
BATCH_SIZE: 2
|
|
||||||
BBOX_FILE: ''
|
|
||||||
BBOX_THRE: 1.0
|
|
||||||
DETECTOR: fpn_dcn
|
|
||||||
DETECTOR_DIR: ''
|
|
||||||
IMAGE_THRE: 0.1
|
|
||||||
IN_VIS_THRE: 0.0
|
|
||||||
MATCH_IOU_THRE: 0.3
|
|
||||||
MODEL_FILE: ''
|
|
||||||
NMS_THRE: 0.6
|
|
||||||
OKS_THRE: 0.5
|
|
||||||
POST_PROCESS: false
|
|
||||||
USE_GT_BBOX: true
|
|
||||||
DEBUG:
|
|
||||||
DEBUG: true
|
|
||||||
SAVE_BATCH_IMAGES_GT: true
|
|
||||||
SAVE_BATCH_IMAGES_PRED: true
|
|
||||||
SAVE_HEATMAPS_GT: true
|
|
||||||
SAVE_HEATMAPS_PRED: true
|
|
Загрузка…
Ссылка в новой задаче