a b/configs/gaitedge/phase2_gaitedge.yaml 

  1
  
data_cfg:





  
  

  2
  
  dataset_name: CASIA-B*





  
  

  3
  
  dataset_root: your_path





  
  

  4
  
  data_in_use: [false, true, true, true]





  
  

  5
  
  dataset_partition: ./datasets/CASIA-B/CASIA-B.json





  
  

  6
  
  num_workers: 1





  
  

  7
  
  remove_no_gallery: false # Remove probe if no gallery for it





  
  

  8
  
  test_dataset_name: CASIA-B





  
  

  9
  
  cache: false





  
  

  10
  





  
  

  11
  
evaluator_cfg:





  
  

  12
  
  enable_float16: true





  
  

  13
  
  restore_ckpt_strict: true





  
  

  14
  
  restore_hint: 20000





  
  

  15
  
  save_name: GaitEdge





  
  

  16
  
  sampler:





  
  

  17
  
    batch_size: 4





  
  

  18
  
    sample_type: all_ordered





  
  

  19
  
    type: InferenceSampler





  
  

  20
  
    frames_all_limit: 512 # limit the number of sampled frames to prevent out of memory





  
  

  21
  
  metric: euc # cos





  
  

  22
  
  transform:





  
  

  23
  
    - type: NoOperation





  
  

  24
  
    - type: BaseRgbTransform





  
  

  25
  
    - type: BaseSilTransform





  
  

  26
  





  
  

  27
  
loss_cfg:





  
  

  28
  
  - loss_term_weight: 1





  
  

  29
  
    margin: 0.2





  
  

  30
  
    type: TripletLoss





  
  

  31
  
    log_prefix: triplet





  
  

  32
  
  - loss_term_weight: 10





  
  

  33
  
    type: BinaryCrossEntropyLoss





  
  

  34
  
    log_prefix: bce





  
  

  35
  
  - loss_term_weight: 1





  
  

  36
  
    scale: 1





  
  

  37
  
    type: CrossEntropyLoss





  
  

  38
  
    log_accuracy: true





  
  

  39
  
    label_smooth: false





  
  

  40
  
    log_prefix: softmax





  
  

  41
  





  
  

  42
  
model_cfg:





  
  

  43
  
  model: GaitEdge





  
  

  44
  
  GaitGL:





  
  

  45
  
    channels: [32, 64, 128]





  
  

  46
  
    class_num: 74





  
  

  47
  
  Segmentation:





  
  

  48
  
    type: U_Net





  
  

  49
  
    in_channels: 3





  
  

  50
  
    freeze_half: true





  
  

  51
  
  edge: true





  
  

  52
  
  align: true





  
  

  53
  
  seg_lr: 0.00001





  
  

  54
  
  kernel_size: 3





  
  

  55
  





  
  

  56
  
optimizer_cfg:





  
  

  57
  
  lr: 0.0001





  
  

  58
  
  solver: Adam





  
  

  59
  
  weight_decay: 0.0005





  
  

  60
  
scheduler_cfg:





  
  

  61
  
  gamma: 0.1





  
  

  62
  
  milestones: # Learning Rate Reduction at each milestones





  
  

  63
  
    - 10000





  
  

  64
  
  scheduler: MultiStepLR





  
  

  65
  
trainer_cfg:





  
  

  66
  
  enable_float16: true # half_percesion float for memory reduction and speedup





  
  

  67
  
  with_test: true





  
  

  68
  
  log_iter: 100





  
  

  69
  
  restore_ckpt_strict: false





  
  

  70
  
  optimizer_reset: true





  
  

  71
  
  scheduler_reset: true





  
  

  72
  
  sync_BN: true





  
  

  73
  
  restore_hint: 





  
  

  74
  
    - Segmentation-25000.pt





  
  

  75
  
    - GaitGL-80000.pt





  
  

  76
  
  save_iter: 2000





  
  

  77
  
  save_name: GaitEdge





  
  

  78
  
  total_iter: 20000





  
  

  79
  
  sampler:





  
  

  80
  
    batch_shuffle: true





  
  

  81
  
    batch_size:





  
  

  82
  
      - 8 # TripletSampler, batch_size[0] indicates Number of Identity





  
  

  83
  
      - 8 #                 batch_size[1] indicates Samples sequqnce for each Identity





  
  

  84
  
    frames_num_fixed: 30





  
  

  85
  
    frames_skip_num: 0





  
  

  86
  
    sample_type: fixed_ordered





  
  

  87
  
    type: TripletSampler





  
  

  88
  
  transform:





  
  

  89
  
    - type: NoOperation





  
  

  90
  
    - type: BaseRgbTransform





  
  

  91
  
    - type: BaseSilTransform