This repository has been archived by the owner on Jun 13, 2024. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 604
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add support for caffe-style ResNet and more flexible pretrained model…
… loading (#51) * add support for resnetv1 * fix ordereddict manipulation * add caffe-style resnet results in README * update cfg files for caffe-style resnet
- Loading branch information
1 parent
7efb8ec
commit 2d723e3
Showing
6 changed files
with
309 additions
and
2 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
77 changes: 77 additions & 0 deletions
77
experiments/coco/resnet101/256x192_d256x3_adam_lr1e-3_caffe.yaml
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,77 @@ | ||
GPUS: '0' | ||
DATA_DIR: '' | ||
OUTPUT_DIR: 'output' | ||
LOG_DIR: 'log' | ||
WORKERS: 4 | ||
PRINT_FREQ: 100 | ||
|
||
DATASET: | ||
DATASET: 'coco' | ||
ROOT: 'data/coco/' | ||
TEST_SET: 'val2017' | ||
TRAIN_SET: 'train2017' | ||
FLIP: true | ||
ROT_FACTOR: 40 | ||
SCALE_FACTOR: 0.3 | ||
MODEL: | ||
NAME: 'pose_resnet' | ||
PRETRAINED: 'models/pytorch/imagenet/resnet101-caffe.pth.tar' | ||
STYLE: 'caffe' | ||
IMAGE_SIZE: | ||
- 192 | ||
- 256 | ||
NUM_JOINTS: 17 | ||
EXTRA: | ||
TARGET_TYPE: 'gaussian' | ||
HEATMAP_SIZE: | ||
- 48 | ||
- 64 | ||
SIGMA: 2 | ||
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: 101 | ||
LOSS: | ||
USE_TARGET_WEIGHT: true | ||
TRAIN: | ||
BATCH_SIZE: 32 | ||
SHUFFLE: true | ||
BEGIN_EPOCH: 0 | ||
END_EPOCH: 140 | ||
RESUME: false | ||
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: 32 | ||
COCO_BBOX_FILE: 'data/coco/person_detection_results/COCO_val2017_detections_AP_H_56_person.json' | ||
BBOX_THRE: 1.0 | ||
FLIP_TEST: false | ||
IMAGE_THRE: 0.0 | ||
IN_VIS_THRE: 0.2 | ||
MODEL_FILE: '' | ||
NMS_THRE: 1.0 | ||
OKS_THRE: 0.9 | ||
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 |
77 changes: 77 additions & 0 deletions
77
experiments/coco/resnet152/256x192_d256x3_adam_lr1e-3_caffe.yaml
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,77 @@ | ||
GPUS: '0' | ||
DATA_DIR: '' | ||
OUTPUT_DIR: 'output' | ||
LOG_DIR: 'log' | ||
WORKERS: 4 | ||
PRINT_FREQ: 100 | ||
|
||
DATASET: | ||
DATASET: 'coco' | ||
ROOT: 'data/coco/' | ||
TEST_SET: 'val2017' | ||
TRAIN_SET: 'train2017' | ||
FLIP: true | ||
ROT_FACTOR: 40 | ||
SCALE_FACTOR: 0.3 | ||
MODEL: | ||
NAME: 'pose_resnet' | ||
PRETRAINED: 'models/pytorch/imagenet/resnet152-caffe.pth.tar' | ||
STYLE: 'caffe' | ||
IMAGE_SIZE: | ||
- 192 | ||
- 256 | ||
NUM_JOINTS: 17 | ||
EXTRA: | ||
TARGET_TYPE: 'gaussian' | ||
HEATMAP_SIZE: | ||
- 48 | ||
- 64 | ||
SIGMA: 2 | ||
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: 32 | ||
SHUFFLE: true | ||
BEGIN_EPOCH: 0 | ||
END_EPOCH: 140 | ||
RESUME: false | ||
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: 32 | ||
COCO_BBOX_FILE: 'data/coco/person_detection_results/COCO_val2017_detections_AP_H_56_person.json' | ||
BBOX_THRE: 1.0 | ||
FLIP_TEST: false | ||
IMAGE_THRE: 0.0 | ||
IN_VIS_THRE: 0.2 | ||
MODEL_FILE: '' | ||
NMS_THRE: 1.0 | ||
OKS_THRE: 0.9 | ||
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 |
77 changes: 77 additions & 0 deletions
77
experiments/coco/resnet50/256x192_d256x3_adam_lr1e-3_caffe.yaml
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,77 @@ | ||
GPUS: '0' | ||
DATA_DIR: '' | ||
OUTPUT_DIR: 'output' | ||
LOG_DIR: 'log' | ||
WORKERS: 4 | ||
PRINT_FREQ: 100 | ||
|
||
DATASET: | ||
DATASET: 'coco' | ||
ROOT: 'data/coco/' | ||
TEST_SET: 'val2017' | ||
TRAIN_SET: 'train2017' | ||
FLIP: true | ||
ROT_FACTOR: 40 | ||
SCALE_FACTOR: 0.3 | ||
MODEL: | ||
NAME: 'pose_resnet' | ||
PRETRAINED: 'models/pytorch/imagenet/resnet50-caffe.pth.tar' | ||
STYLE: 'caffe' | ||
IMAGE_SIZE: | ||
- 192 | ||
- 256 | ||
NUM_JOINTS: 17 | ||
EXTRA: | ||
TARGET_TYPE: 'gaussian' | ||
HEATMAP_SIZE: | ||
- 48 | ||
- 64 | ||
SIGMA: 2 | ||
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: 32 | ||
SHUFFLE: true | ||
BEGIN_EPOCH: 0 | ||
END_EPOCH: 140 | ||
RESUME: false | ||
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: 32 | ||
COCO_BBOX_FILE: 'data/coco/person_detection_results/COCO_val2017_detections_AP_H_56_person.json' | ||
BBOX_THRE: 1.0 | ||
FLIP_TEST: false | ||
IMAGE_THRE: 0.0 | ||
IN_VIS_THRE: 0.2 | ||
MODEL_FILE: '' | ||
NMS_THRE: 1.0 | ||
OKS_THRE: 0.9 | ||
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 |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters