Skip to content

Commit fa0a700

Browse files
committed
add transfer benchmark test configs
1 parent cf8d7c3 commit fa0a700

File tree

4 files changed

+99
-0
lines changed

4 files changed

+99
-0
lines changed

experiments/unihcp/release/batch_test.sh

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,9 @@ g=$((${gpus}<8?${gpus}:8))
2525
declare -A test_info_list
2626
test_info_list[pose_lpe]=0
2727
test_info_list[ochuman_pose]=0
28+
#test_info_list[pose_mpii_lpe]=x # requires mpii queries
2829
test_info_list[par_lpe]=1
30+
#test_info_list[par_atr_lpe]=x # requires ATR queries
2931
test_info_list[reid]=2
3032
test_info_list[reid_cuhk3]=2
3133
test_info_list[reid_duke]=2
@@ -35,6 +37,7 @@ test_info_list[par_lip_lpe]=3
3537
test_info_list[par_cihp_lpe]=4
3638
test_info_list[pa100k_lpe]=5
3739
test_info_list[rap2_lpe]=6
40+
#test_info_list[peta_lpe]=x # requires PETA queries
3841
test_info_list[pose_aic_lpe]=7
3942
test_info_list[peddet_caltech]=8
4043
test_info_list[peddet_inter_lpe]=8
Lines changed: 41 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,41 @@
1+
common:
2+
backbone:
3+
kwargs:
4+
test_pos_mode: learnable_interpolate
5+
tester:
6+
type: TesterMultiTaskDev
7+
8+
sync: True
9+
collate: dev
10+
11+
# task_specific_param = ['backbone', 'neck', 'decoder', 'dataset', 'sampler', 'lr_scheduler', 'optimizer']
12+
tasks:
13+
0:
14+
name: ParTest
15+
loss_weight: 1.0 # *redundant*
16+
gres_ratio: 1 # int, > 0| world/sum*ratio, *redundant*
17+
dataset:
18+
type: ATRParsingDataset
19+
kwargs:
20+
data_path: /mnt/path...to.../ATR
21+
dataset: val
22+
is_train: False
23+
cfg:
24+
# mean: [ 0.485, 0.456, 0.406 ]
25+
# std: [ 0.229, 0.224, 0.225 ]
26+
eval_crop_size: [ 480, 480 ]
27+
is_flip: False
28+
is_multi_scale: False
29+
30+
ignore_value: 255 # TODO: duplicated with decoder.kwargs.ignore_value
31+
num_classes: 18
32+
label_list: [ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, ]
33+
34+
sampler:
35+
batch_size: 16 # per card
36+
37+
38+
# extra:
39+
# min_sizes: [320, 480, 640, 800, 960, 1120] # TTA
40+
# max_size: 4480
41+
# flip: True
Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,26 @@
1+
common:
2+
backbone:
3+
kwargs:
4+
test_pos_mode: learnable_interpolate
5+
tester:
6+
type: TesterMultiTaskDev
7+
8+
sync: True
9+
collate: dev
10+
11+
tasks:
12+
0:
13+
name: pedattrTest
14+
loss_weight: 1.0
15+
gres_ratio: 1 # int, > 0| world/sum*ratio
16+
dataset:
17+
type: AttrDataset
18+
kwargs:
19+
task_spec:
20+
dataset: 'peta'
21+
data_path: /mnt/path...to.../peta/dataset.pkl
22+
root_path: /mnt/path...to.../peta/images/
23+
augmentation:
24+
height: 256
25+
width: 192
26+
train: False
Lines changed: 29 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,29 @@
1+
common:
2+
backbone:
3+
kwargs:
4+
test_pos_mode: learnable_interpolate
5+
tester:
6+
type: TesterMultiTaskDev
7+
8+
sync: True
9+
collate: dev_collate
10+
11+
# task_specific_param = ['backbone', 'neck', 'decoder', 'dataset', 'sampler', 'lr_scheduler', 'optimizer']
12+
tasks:
13+
0:
14+
name: MPIIPoseTest # deprecated
15+
loss_weight: 1.0 # *redundant*
16+
gres_ratio: 1 # int, > 0| world/sum*ratio, *redundant*
17+
dataset:
18+
type: MPIIPosDatasetDev
19+
kwargs:
20+
ann_file: '/mnt/path...to.../coco/annotations/mpii_val.json'
21+
img_prefix: '/mnt/path...to.../MPI/images/'
22+
test_mode: True
23+
sampler:
24+
batch_size: 256 # per card
25+
evaluation:
26+
cfg:
27+
interval: 10
28+
metric: PCKh
29+

0 commit comments

Comments
 (0)