Clear up the confusion about gait3d.

This commit is contained in:
darkliang
2023-04-06 12:45:50 +08:00
parent 37258d149f
commit ef00896028
4 changed files with 5 additions and 8 deletions
+1 -2
View File
@@ -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
+2 -2
View File
@@ -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
+1 -3
View File
@@ -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:
+1 -1
View File
@@ -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`