Skip to content

Commit 2c29afa

Browse files
jdyjjjjdyjjj
authored
Support skeleton (#155)
* pose * pose * pose * pose * 你的提交消息 * pose * pose * Delete train1.sh * pretreatment * configs * pose * reference * Update gaittr.py * naming * naming * Update transform.py * update for datasets * update README * update name and README * update * Update transform.py
1 parent 853bb18 commit 2c29afa

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

41 files changed

+4251
-12
lines changed

‎configs/gaitgraph1/gaitgraph1_phase1.yaml

+98
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,98 @@
1+
data_cfg:
2+
dataset_name: CASIA-B
3+
dataset_root: your_path
4+
dataset_partition: ./datasets/CASIA-B/CASIA-B.json
5+
test_dataset_name: CASIA-B
6+
num_workers: 8
7+
remove_no_gallery: false
8+
frame_threshold: 60
9+
10+
evaluator_cfg:
11+
enable_float16: false
12+
restore_ckpt_strict: true
13+
restore_hint: 6000
14+
save_name: GaitGraph1_phase1
15+
sampler:
16+
batch_size: 256
17+
frames_num_fixed: 60
18+
frames_num_max: 50
19+
frames_num_min: 25
20+
sample_type: fixed_ordered
21+
frames_skip_num: 0
22+
metric: euc
23+
eval_func: evaluate_indoor_dataset
24+
transform:
25+
- type: Compose
26+
trf_cfg:
27+
- type: GaitGraph1Input
28+
29+
loss_cfg:
30+
- loss_term_weight: 1
31+
temperature: 0.01
32+
type: SupConLoss_Re
33+
log_prefix: SupConLoss
34+
35+
model_cfg:
36+
model: GaitGraph1
37+
joint_format: coco
38+
input_num: 1
39+
reduction: 8
40+
block: Bottleneck # Basic, initial
41+
input_branch:
42+
- 3
43+
- 64
44+
- 64
45+
- 32
46+
main_stream:
47+
- 32
48+
- 128
49+
- 128
50+
- 256
51+
- 256
52+
num_class: 128
53+
tta: true
54+
55+
optimizer_cfg:
56+
lr: 0.01
57+
solver: Adam
58+
weight_decay: 0.00001
59+
60+
scheduler_cfg:
61+
max_lr: 0.01
62+
total_steps: 6000
63+
scheduler: OneCycleLR
64+
div_factor: 25
65+
66+
trainer_cfg:
67+
enable_float16: false
68+
log_iter: 1000
69+
with_test: true
70+
restore_ckpt_strict: false
71+
restore_hint: 0
72+
save_iter: 1000
73+
save_name: GaitGraph1_phase1
74+
sync_BN: true
75+
total_iter: 6000
76+
sampler:
77+
batch_shuffle: true
78+
frames_num_fixed: 100
79+
frames_num_max: 50
80+
frames_num_min: 25
81+
sample_type: fixed_ordered #Repeat sample
82+
frames_skip_num: 0
83+
batch_size: 128
84+
type: CommonSampler
85+
transform:
86+
- type: TwoView
87+
trf_cfg:
88+
- type: InversePosesPre
89+
probability: 0.1
90+
- type: FlipSequence
91+
probability: 0.5
92+
- type: RandomSelectSequence
93+
sequence_length: 60
94+
- type: PointNoise
95+
std: 0.05
96+
- type: JointNoise
97+
std: 0.2
98+
- type: GaitGraph1Input

‎configs/gaitgraph1/gaitgraph1_phase1_GREW.yaml

+100
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,100 @@
1+
data_cfg:
2+
dataset_name: GREW
3+
dataset_root: your_path
4+
dataset_partition: ./datasets/GREW/GREW.json
5+
test_dataset_name: GREW
6+
num_workers: 8
7+
remove_no_gallery: false
8+
frame_threshold: 16
9+
10+
evaluator_cfg:
11+
enable_float16: false
12+
restore_ckpt_strict: true
13+
restore_hint: 150000
14+
save_name: GaitGraph1_phase1
15+
sampler:
16+
batch_size: 256
17+
frames_num_fixed: 501
18+
frames_num_max: 50
19+
frames_num_min: 25
20+
sample_type: fixed_ordered
21+
frames_skip_num: 0
22+
metric: euc
23+
# eval_func: GREW_submission
24+
eval_func: evaluate_real_scene
25+
transform:
26+
- type: Compose
27+
trf_cfg:
28+
- type: SelectSequenceCenter
29+
sequence_length: 16
30+
- type: GaitGraph1Input
31+
32+
loss_cfg:
33+
- loss_term_weight: 1
34+
temperature: 0.01
35+
type: SupConLoss_Re
36+
log_prefix: SupConLoss
37+
38+
model_cfg:
39+
model: GaitGraph1
40+
joint_format: coco
41+
input_num: 1
42+
reduction: 8
43+
block: Bottleneck # Basic, initial
44+
input_branch:
45+
- 3
46+
- 64
47+
- 64
48+
- 32
49+
main_stream:
50+
- 32
51+
- 128
52+
- 128
53+
- 256
54+
- 256
55+
num_class: 256
56+
tta: true
57+
58+
optimizer_cfg:
59+
lr: 0.01
60+
solver: Adam
61+
weight_decay: 0.00001
62+
63+
scheduler_cfg:
64+
max_lr: 0.01
65+
total_steps: 150000
66+
scheduler: OneCycleLR
67+
68+
trainer_cfg:
69+
enable_float16: false
70+
log_iter: 100
71+
with_test: true
72+
restore_ckpt_strict: false
73+
restore_hint: 0
74+
save_iter: 5000
75+
save_name: GaitGraph1_phase1
76+
sync_BN: true
77+
total_iter: 150000
78+
sampler:
79+
batch_shuffle: true
80+
frames_num_fixed: 501
81+
frames_num_max: 50
82+
frames_num_min: 25
83+
sample_type: fixed_ordered #Repeat sample
84+
frames_skip_num: 0
85+
batch_size: 128
86+
type: CommonSampler
87+
transform:
88+
- type: TwoView
89+
trf_cfg:
90+
- type: MirrorPoses
91+
probability: 0.5
92+
- type: FlipSequence
93+
probability: 0.5
94+
- type: RandomSelectSequence
95+
sequence_length: 16
96+
- type: PointNoise
97+
std: 0.05
98+
- type: JointNoise
99+
std: 0.1
100+
- type: GaitGraph1Input

‎configs/gaitgraph1/gaitgraph1_phase1_Gait3D.yaml

+100
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,100 @@
1+
data_cfg:
2+
dataset_name: Gait3D
3+
dataset_root: your_path
4+
dataset_partition: ./datasets/Gait3D/Gait3D.json
5+
test_dataset_name: Gait3D
6+
num_workers: 8
7+
remove_no_gallery: false
8+
frame_threshold: 24
9+
10+
evaluator_cfg:
11+
enable_float16: false
12+
restore_ckpt_strict: true
13+
restore_hint: 20000
14+
save_name: GaitGraph1_phase1
15+
sampler:
16+
batch_size: 256
17+
frames_num_fixed: 130
18+
frames_num_max: 50
19+
frames_num_min: 25
20+
sample_type: fixed_ordered
21+
frames_skip_num: 0
22+
metric: euc
23+
eval_func: evaluate_Gait3D
24+
transform:
25+
- type: Compose
26+
trf_cfg:
27+
- type: SelectSequenceCenter
28+
sequence_length: 24
29+
- type: GaitGraph1Input
30+
31+
loss_cfg:
32+
- loss_term_weight: 1
33+
temperature: 0.01
34+
type: SupConLoss_Re
35+
log_prefix: SupConLoss
36+
37+
model_cfg:
38+
model: GaitGraph1
39+
joint_format: coco
40+
input_num: 1
41+
reduction: 8
42+
block: Bottleneck # Basic, initial
43+
input_branch:
44+
- 3
45+
- 64
46+
- 64
47+
- 32
48+
main_stream:
49+
- 32
50+
- 128
51+
- 128
52+
- 256
53+
- 256
54+
num_class: 128
55+
tta: true
56+
57+
optimizer_cfg:
58+
lr: 0.01
59+
solver: Adam
60+
weight_decay: 0.00001
61+
62+
scheduler_cfg:
63+
max_lr: 0.01
64+
total_steps: 20000
65+
scheduler: OneCycleLR
66+
67+
trainer_cfg:
68+
enable_float16: false
69+
log_iter: 100
70+
with_test: true
71+
restore_ckpt_strict: false
72+
restore_hint: 0
73+
save_iter: 500
74+
save_name: GaitGraph1_phase1
75+
sync_BN: true
76+
total_iter: 20000
77+
sampler:
78+
batch_shuffle: true
79+
frames_num_fixed: 130
80+
frames_num_max: 50
81+
frames_num_min: 25
82+
sample_type: fixed_ordered #Repeat sample
83+
frames_skip_num: 0
84+
batch_size: 128
85+
type: CommonSampler
86+
transform:
87+
- type: TwoView
88+
trf_cfg:
89+
- type: MirrorPoses
90+
probability: 0.5
91+
- type: FlipSequence
92+
probability: 0.5
93+
- type: RandomSelectSequence
94+
sequence_length: 16
95+
- type: PointNoise
96+
std: 0.05
97+
- type: JointNoise
98+
std: 0.1
99+
- type: GaitGraph1Input
100+

0 commit comments

Comments
 (0)