a | b/configs/gaitpart/gaitpart.yaml | ||
---|---|---|---|
1 | data_cfg: |
||
2 | dataset_name: CASIA-B |
||
3 | dataset_root: your_path |
||
4 | num_workers: 1 |
||
5 | remove_no_gallery: false |
||
6 | test_dataset_name: CASIA-B |
||
7 | |||
8 | evaluator_cfg: |
||
9 | enable_float16: false |
||
10 | restore_ckpt_strict: true |
||
11 | restore_hint: 120000 |
||
12 | save_name: GaitPart |
||
13 | sampler: |
||
14 | batch_size: 16 |
||
15 | sample_type: all_ordered |
||
16 | type: InferenceSampler |
||
17 | metric: euc # cos |
||
18 | |||
19 | loss_cfg: |
||
20 | loss_term_weight: 1.0 |
||
21 | margin: 0.2 |
||
22 | type: TripletLoss |
||
23 | log_prefix: triplet |
||
24 | |||
25 | model_cfg: |
||
26 | model: GaitPart |
||
27 | backbone_cfg: |
||
28 | in_channels: 1 |
||
29 | layers_cfg: |
||
30 | - BC-32 |
||
31 | - BC-32 |
||
32 | - M |
||
33 | - FC-64-2 |
||
34 | - FC-64-2 |
||
35 | - M |
||
36 | - FC-128-3 |
||
37 | - FC-128-3 |
||
38 | type: Plain |
||
39 | SeparateFCs: |
||
40 | in_channels: 128 |
||
41 | out_channels: 128 |
||
42 | parts_num: 16 |
||
43 | bin_num: |
||
44 | - 16 |
||
45 | |||
46 | |||
47 | optimizer_cfg: |
||
48 | lr: 0.0001 |
||
49 | momentum: 0.9 |
||
50 | solver: Adam |
||
51 | weight_decay: 0.0 |
||
52 | |||
53 | scheduler_cfg: |
||
54 | gamma: 0.1 |
||
55 | milestones: |
||
56 | - 100000 |
||
57 | scheduler: MultiStepLR |
||
58 | |||
59 | trainer_cfg: |
||
60 | enable_float16: true |
||
61 | log_iter: 100 |
||
62 | with_test: true |
||
63 | restore_ckpt_strict: true |
||
64 | restore_hint: 0 |
||
65 | save_iter: 10000 |
||
66 | save_name: GaitPart |
||
67 | sync_BN: false |
||
68 | total_iter: 120000 |
||
69 | sampler: |
||
70 | batch_shuffle: false |
||
71 | batch_size: |
||
72 | - 8 |
||
73 | - 16 |
||
74 | frames_num_fixed: 30 |
||
75 | frames_num_max: 50 |
||
76 | frames_num_min: 25 |
||
77 | frames_skip_num: 10 |
||
78 | sample_type: fixed_ordered |
||
79 | type: TripletSampler |