Clear up the confusion about gait3d.
This commit is contained in:
@@ -9,9 +9,8 @@ data_cfg:
|
|||||||
evaluator_cfg:
|
evaluator_cfg:
|
||||||
enable_float16: true
|
enable_float16: true
|
||||||
restore_ckpt_strict: true
|
restore_ckpt_strict: true
|
||||||
restore_hint: 0
|
restore_hint: 60000
|
||||||
save_name: GaitBase_DA
|
save_name: GaitBase_DA
|
||||||
#eval_func: GREW_submission
|
|
||||||
sampler:
|
sampler:
|
||||||
batch_shuffle: false
|
batch_shuffle: false
|
||||||
batch_size: 16
|
batch_size: 16
|
||||||
|
|||||||
@@ -19,7 +19,7 @@ evaluator_cfg:
|
|||||||
frames_all_limit: 720 # limit the number of sampled frames to prevent out of memory
|
frames_all_limit: 720 # limit the number of sampled frames to prevent out of memory
|
||||||
metric: euc # cos
|
metric: euc # cos
|
||||||
transform:
|
transform:
|
||||||
- type: BaseSilTransform
|
- type: BaseSilCuttingTransform
|
||||||
|
|
||||||
loss_cfg:
|
loss_cfg:
|
||||||
- loss_term_weight: 1.0
|
- loss_term_weight: 1.0
|
||||||
@@ -103,7 +103,7 @@ trainer_cfg:
|
|||||||
trf_cfg:
|
trf_cfg:
|
||||||
- type: RandomPerspective
|
- type: RandomPerspective
|
||||||
prob: 0.2
|
prob: 0.2
|
||||||
- type: BaseSilTransform
|
- type: BaseSilCuttingTransform
|
||||||
- type: RandomHorizontalFlip
|
- type: RandomHorizontalFlip
|
||||||
prob: 0.2
|
prob: 0.2
|
||||||
- type: RandomRotate
|
- type: RandomRotate
|
||||||
|
|||||||
@@ -11,12 +11,10 @@ evaluator_cfg:
|
|||||||
restore_ckpt_strict: true
|
restore_ckpt_strict: true
|
||||||
restore_hint: 120000
|
restore_hint: 120000
|
||||||
save_name: GaitBase
|
save_name: GaitBase
|
||||||
#eval_func: GREW_submission
|
|
||||||
sampler:
|
sampler:
|
||||||
batch_shuffle: false
|
batch_shuffle: false
|
||||||
batch_size: 16
|
batch_size: 16
|
||||||
sample_type: all_ordered # all indicates whole sequence used to test, while ordered means input sequence by its natural order;
|
sample_type: all_ordered # all indicates whole sequence used to test, while ordered means input sequence by its natural order; Other options: fixed_unordered
|
||||||
Other options: fixed_unordered
|
|
||||||
frames_all_limit: 720 # limit the number of sampled frames to prevent out of memory
|
frames_all_limit: 720 # limit the number of sampled frames to prevent out of memory
|
||||||
metric: euc # cos
|
metric: euc # cos
|
||||||
transform:
|
transform:
|
||||||
|
|||||||
@@ -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'
|
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
|
## Train
|
||||||
### Baseline model:
|
### 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`
|
`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`
|
||||||
|
|||||||
Reference in New Issue
Block a user