Clear up the confusion about gait3d.

cocoJennie · Apr 6, 2023 · ef00896 · ef00896
1 parent 37258d1
commit ef00896
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 8 deletions.
diff --git a/configs/gaitbase/gaitbase_da_casiab.yaml b/configs/gaitbase/gaitbase_da_casiab.yaml
@@ -9,9 +9,8 @@ data_cfg:
 evaluator_cfg:
   enable_float16: true
   restore_ckpt_strict: true
-  restore_hint: 0
+  restore_hint: 60000
   save_name: GaitBase_DA
-  #eval_func: GREW_submission
   sampler:
     batch_shuffle: false
     batch_size: 16

diff --git a/configs/gaitbase/gaitbase_da_gait3d.yaml b/configs/gaitbase/gaitbase_da_gait3d.yaml
@@ -19,7 +19,7 @@ evaluator_cfg:
     frames_all_limit: 720 # limit the number of sampled frames to prevent out of memory
   metric: euc # cos
   transform:
-    - type: BaseSilTransform
+    - type: BaseSilCuttingTransform
 
 loss_cfg:
   - loss_term_weight: 1.0
@@ -103,7 +103,7 @@ trainer_cfg:
       trf_cfg:
         - type: RandomPerspective
           prob: 0.2
-        - type: BaseSilTransform
+        - type: BaseSilCuttingTransform
         - type: RandomHorizontalFlip
           prob: 0.2
         - type: RandomRotate

diff --git a/configs/gaitbase/gaitbase_oumvlp.yaml b/configs/gaitbase/gaitbase_oumvlp.yaml
@@ -11,12 +11,10 @@ evaluator_cfg:
   restore_ckpt_strict: true
   restore_hint: 120000
   save_name: GaitBase
-  #eval_func: GREW_submission
   sampler:
     batch_shuffle: false
     batch_size: 16
-    sample_type: all_ordered # all indicates whole sequence used to test, while ordered means input sequence by its natural order; 
-Other options:   fixed_unordered
+    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
   metric: euc # cos
   transform:

diff --git a/datasets/Gait3D/README.md b/datasets/Gait3D/README.md
@@ -11,7 +11,7 @@ python datasets/pretreatment_smpl.py --input_path 'Gait3D/3D_SMPLs' --output_pat
 
 python datasets/Gait3D/merge_two_modality.py --sils_path 'Gait3D-sils-64-64-pkl' --smpls_path 'Gait3D-smpls-pkl' --output_path 'Gait3D-merged-pkl' --link 'hard'
 ```
-**Note**: If you use the processed pickle files directly, then the size of silhouette is `64x44`, which means that the pixels on both sides of the horizontal direction can no longer be cut when transforming.
+
 ## Train
 ### Baseline model:
 `CUDA_VISIBLE_DEVICES=0,1,2,3 python -m torch.distributed.launch --nproc_per_node=4 opengait/main.py --cfgs ./configs/baseline/baseline_Gait3D.yaml --phase train`