Datasets
Models
Diff of
/configs/gaitgraph2/gaitgraph2_CCPG.yaml
[000000]
..
[fd9ef4]
Switch to unified view
| a | b/configs/gaitgraph2/gaitgraph2_CCPG.yaml | ||
|---|---|---|---|
| 1 | data_cfg: |
||
| 2 | dataset_name: CCPG |
||
| 3 | dataset_root: your_path |
||
| 4 | dataset_partition: ./datasets/CCPG/CCPG.json |
||
| 5 | test_dataset_name: CCPG |
||
| 6 | num_workers: 8 |
||
| 7 | remove_no_gallery: false |
||
| 8 | |||
| 9 | evaluator_cfg: |
||
| 10 | enable_float16: false |
||
| 11 | restore_ckpt_strict: true |
||
| 12 | restore_hint: 2000 |
||
| 13 | save_name: GaitGraph2_CCPG |
||
| 14 | sampler: |
||
| 15 | batch_size: 256 |
||
| 16 | frames_num_fixed: 30 |
||
| 17 | frames_num_max: 50 |
||
| 18 | frames_num_min: 25 |
||
| 19 | sample_type: fixed_ordered |
||
| 20 | frames_skip_num: 0 |
||
| 21 | metric: cos |
||
| 22 | eval_func: evaluate_CCPG |
||
| 23 | transform: |
||
| 24 | - type: Compose |
||
| 25 | trf_cfg: |
||
| 26 | - type: NormalizeEmpty |
||
| 27 | - type: GaitGraphMultiInput |
||
| 28 | joint_format: alphapose |
||
| 29 | |||
| 30 | |||
| 31 | loss_cfg: |
||
| 32 | - loss_term_weight: 1 |
||
| 33 | temperature: 0.01 |
||
| 34 | type: SupConLoss_Lp |
||
| 35 | log_prefix: SupConLoss |
||
| 36 | |||
| 37 | model_cfg: |
||
| 38 | model: GaitGraph2 |
||
| 39 | joint_format: alphapose |
||
| 40 | input_num: 3 |
||
| 41 | reduction: 4 |
||
| 42 | block: Bottleneck # Basic, initial |
||
| 43 | input_branch: |
||
| 44 | - 5 |
||
| 45 | - 64 |
||
| 46 | - 32 |
||
| 47 | main_stream: |
||
| 48 | - 32 |
||
| 49 | - 128 |
||
| 50 | - 256 |
||
| 51 | num_class: 128 |
||
| 52 | tta: true |
||
| 53 | |||
| 54 | optimizer_cfg: |
||
| 55 | lr: 0.005 |
||
| 56 | solver: Adam |
||
| 57 | weight_decay: 0.00001 |
||
| 58 | |||
| 59 | scheduler_cfg: |
||
| 60 | three_phase: True |
||
| 61 | max_lr: 0.005 |
||
| 62 | total_steps: 2000 |
||
| 63 | scheduler: OneCycleLR |
||
| 64 | |||
| 65 | trainer_cfg: |
||
| 66 | enable_float16: false |
||
| 67 | log_iter: 20 |
||
| 68 | with_test: true |
||
| 69 | restore_ckpt_strict: false |
||
| 70 | restore_hint: 0 |
||
| 71 | save_iter: 200 |
||
| 72 | save_name: GaitGraph2 |
||
| 73 | sync_BN: true |
||
| 74 | total_iter: 2000 |
||
| 75 | sampler: |
||
| 76 | batch_shuffle: true |
||
| 77 | frames_num_fixed: 60 |
||
| 78 | frames_num_max: 50 |
||
| 79 | frames_num_min: 25 |
||
| 80 | sample_type: fixed_ordered #Repeat sample |
||
| 81 | frames_skip_num: 0 |
||
| 82 | batch_size: 768 |
||
| 83 | type: CommonSampler |
||
| 84 | |||
| 85 | transform: |
||
| 86 | - type: Compose |
||
| 87 | trf_cfg: |
||
| 88 | - type: NormalizeEmpty |
||
| 89 | - type: FlipSequence |
||
| 90 | probability: 0.5 |
||
| 91 | - type: InversePosesPre |
||
| 92 | probability: 0.1 |
||
| 93 | joint_format: alphapose |
||
| 94 | - type: JointNoise |
||
| 95 | std: 0.1 |
||
| 96 | - type: PointNoise |
||
| 97 | std: 0.05 |
||
| 98 | - type: RandomMove |
||
| 99 | random_r: |
||
| 100 | - 3 |
||
| 101 | - 1 |
||
| 102 | - type: GaitGraphMultiInput |
||
| 103 | joint_format: alphapose |