Skip to content

Commit f59a6d8

Browse files
longcwlongcw
committed
add aifi dataset
1 parent 3bf0bc4 commit f59a6d8

File tree

5 files changed

+400
-2
lines changed

5 files changed

+400
-2
lines changed

‎experiments/aifi/hrnet/w32_256x192_adam_lr1e-3.yaml‎

Lines changed: 127 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,127 @@
1+
AUTO_RESUME: true
2+
CUDNN:
3+
BENCHMARK: true
4+
DETERMINISTIC: false
5+
ENABLED: true
6+
DATA_DIR: ''
7+
GPUS: (0,)
8+
OUTPUT_DIR: 'output'
9+
LOG_DIR: 'log'
10+
WORKERS: 24
11+
PRINT_FREQ: 100
12+
13+
DATASET:
14+
COLOR_RGB: true
15+
DATA_FORMAT: jpg
16+
FLIP: true
17+
NUM_JOINTS_HALF_BODY: 8
18+
PROB_HALF_BODY: 0.3
19+
ROT_FACTOR: 45
20+
SCALE_FACTOR: 0.35
21+
DATASET: 'aifi'
22+
ROOT: '/media/longc/LSSD/Public/PILSNU'
23+
TEST_SET: 'train'
24+
TRAIN_SET: 'train'
25+
MODEL:
26+
INIT_WEIGHTS: true
27+
NAME: pose_hrnet
28+
NUM_JOINTS: 17
29+
PRETRAINED: 'models/pytorch/imagenet/hrnet_w32-36af842e.pth'
30+
TARGET_TYPE: gaussian
31+
IMAGE_SIZE:
32+
- 192
33+
- 256
34+
HEATMAP_SIZE:
35+
- 48
36+
- 64
37+
SIGMA: 2
38+
EXTRA:
39+
PRETRAINED_LAYERS:
40+
- 'conv1'
41+
- 'bn1'
42+
- 'conv2'
43+
- 'bn2'
44+
- 'layer1'
45+
- 'transition1'
46+
- 'stage2'
47+
- 'transition2'
48+
- 'stage3'
49+
- 'transition3'
50+
- 'stage4'
51+
FINAL_CONV_KERNEL: 1
52+
STAGE2:
53+
NUM_MODULES: 1
54+
NUM_BRANCHES: 2
55+
BLOCK: BASIC
56+
NUM_BLOCKS:
57+
- 4
58+
- 4
59+
NUM_CHANNELS:
60+
- 32
61+
- 64
62+
FUSE_METHOD: SUM
63+
STAGE3:
64+
NUM_MODULES: 4
65+
NUM_BRANCHES: 3
66+
BLOCK: BASIC
67+
NUM_BLOCKS:
68+
- 4
69+
- 4
70+
- 4
71+
NUM_CHANNELS:
72+
- 32
73+
- 64
74+
- 128
75+
FUSE_METHOD: SUM
76+
STAGE4:
77+
NUM_MODULES: 3
78+
NUM_BRANCHES: 4
79+
BLOCK: BASIC
80+
NUM_BLOCKS:
81+
- 4
82+
- 4
83+
- 4
84+
- 4
85+
NUM_CHANNELS:
86+
- 32
87+
- 64
88+
- 128
89+
- 256
90+
FUSE_METHOD: SUM
91+
LOSS:
92+
USE_TARGET_WEIGHT: true
93+
TRAIN:
94+
BATCH_SIZE_PER_GPU: 32
95+
SHUFFLE: true
96+
BEGIN_EPOCH: 0
97+
END_EPOCH: 210
98+
OPTIMIZER: adam
99+
LR: 0.001
100+
LR_FACTOR: 0.1
101+
LR_STEP:
102+
- 170
103+
- 200
104+
WD: 0.0001
105+
GAMMA1: 0.99
106+
GAMMA2: 0.0
107+
MOMENTUM: 0.9
108+
NESTEROV: false
109+
TEST:
110+
BATCH_SIZE_PER_GPU: 32
111+
COCO_BBOX_FILE: '/extra/code/maskrcnn-benchmark/inference/PILSNU/bbox.json'
112+
BBOX_THRE: 1.0
113+
IMAGE_THRE: 0.0
114+
IN_VIS_THRE: 0.2
115+
MODEL_FILE: ''
116+
NMS_THRE: 1.0
117+
OKS_THRE: 0.9
118+
USE_GT_BBOX: false
119+
FLIP_TEST: true
120+
POST_PROCESS: true
121+
SHIFT_HEATMAP: true
122+
DEBUG:
123+
DEBUG: true
124+
SAVE_BATCH_IMAGES_GT: true
125+
SAVE_BATCH_IMAGES_PRED: true
126+
SAVE_HEATMAPS_GT: true
127+
SAVE_HEATMAPS_PRED: true

‎experiments/aifi/resnet/res50_256x192_d256x3_adam_lr1e-3.yaml‎

Lines changed: 83 additions & 0 deletions
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,)
8+
OUTPUT_DIR: 'output'
9+
LOG_DIR: 'log'
10+
WORKERS: 4
11+
PRINT_FREQ: 100
12+
13+
DATASET:
14+
COLOR_RGB: false
15+
DATASET: 'mot'
16+
ROOT: '/data/MOT16'
17+
TEST_SET: 'train'
18+
TRAIN_SET: 'train'
19+
FLIP: true
20+
ROT_FACTOR: 40
21+
SCALE_FACTOR: 0.3
22+
MODEL:
23+
NAME: 'pose_resnet'
24+
PRETRAINED: 'models/pytorch/imagenet/resnet50-19c8e357.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: 50
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: ''
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: false
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

‎lib/dataset/JointsDataset.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -115,7 +115,7 @@ def __getitem__(self, idx):
115115

116116
image_file = db_rec['image']
117117
image_id = db_rec.get('image_id', -1)
118-
bbox_tlwh = db_rec.get('bbox_tlwh', (0, 0, 0, 0))
118+
bbox_tlwh = db_rec.get('bbox_tlwh', np.zeros(5, dtype=float))
119119

120120
if self.data_format == 'zip':
121121
from utils import zipreader

‎lib/dataset/__init__.py‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,4 +11,5 @@
1111
from .mpii import MPIIDataset as mpii
1212
from .coco import COCODataset as coco
1313
from .posetrack import PoseTrackDataset as posetrack
14-
from .mot import MOTDataset as mot
14+
from .mot import MOTDataset as mot
15+
from .aifi import AIFIDataset as aifi

0 commit comments

Comments
 (0)