--- a +++ b/configs/swingait/swingait3D_B1122C2_CCPG.yaml @@ -0,0 +1,100 @@ +data_cfg: + dataset_name: CCPG + dataset_root: your_path + dataset_partition: ./datasets/CCPG/CCPG.json + num_workers: 1 + data_in_use: [True, False, False, False] + remove_no_gallery: false # Remove probe if no gallery for it + test_dataset_name: CCPG + +evaluator_cfg: + enable_float16: true + restore_ckpt_strict: true + restore_hint: 80000 + save_name: SwinGait3D_B1122_C2 + sampler: + batch_shuffle: false + batch_size: 4 + sample_type: all_ordered # all indicates whole sequence used to test, while ordered means input sequence by its natural order; Other options: fixed_unordered + frames_all_limit: 720 # limit the number of sampled frames to prevent out of memory + eval_func: evaluate_CCPG + metric: euc # cos + transform: + - type: BaseSilCuttingTransform + +loss_cfg: + - loss_term_weight: 1.0 + margin: 0.2 + type: TripletLoss + log_prefix: triplet + - loss_term_weight: 1.0 + scale: 16 + type: CrossEntropyLoss + log_prefix: softmax + log_accuracy: true + +model_cfg: + model: SwinGait + Backbone: + mode: p3d + in_channels: 1 + layers: + - 1 + - 1 + - 2 + - 2 + channels: + - 64 + - 128 + bin_num: + - 15 + SeparateBNNecks: + in_channels: 256 + class_num: 100 + parts_num: 16 + +optimizer_cfg: + lr: 0.0003 + solver: AdamW + weight_decay: 0.02 + +scheduler_cfg: + gamma: 0.1 + milestones: # Learning Rate Reduction at each milestones + - 100000000 + scheduler: MultiStepLR + +trainer_cfg: + enable_float16: true # half_percesion float for memory reduction and speedup + fix_BN: false + log_iter: 100 + with_test: false + optimizer_reset: True + restore_ckpt_strict: false + restore_hint: ./output/CCPG/DeepGaitV2/DeepGaitV2_B1111_C2_P3D/checkpoints/DeepGaitV2_B1111_C2_P3D-60000.pt + save_iter: 20000 + save_name: SwinGait3D_B1122_C2 + sync_BN: true + T_max_iter: 60000 + total_iter: 80000 + sampler: + batch_shuffle: true + batch_size: + - 8 # TripletSampler, batch_size[0] indicates Number of Identity + - 16 # batch_size[1] indicates Samples sequqnce for each Identity + frames_num_fixed: 30 # fixed frames number for training + frames_skip_num: 4 + sample_type: fixed_ordered # fixed control input frames number, unordered for controlling order of input tensor; Other options: unfixed_ordered or all_ordered + type: TripletSampler + transform: + - type: Compose + trf_cfg: + - type: RandomPerspective + prob: 0.2 + - type: BaseSilCuttingTransform + - type: RandomHorizontalFlip + prob: 0.2 + - type: RandomRotate + prob: 0.2 + +