SkeletonGait

This commit is contained in:
Jingzhe Ma
2024-03-07 21:10:22 +08:00
parent 662ffb25b9
commit 29578ccfd7
12 changed files with 59 additions and 29 deletions
@@ -1,6 +1,6 @@
data_cfg:
dataset_name: GREW
dataset_root: your_path
dataset_root: /data3/gait_heatmap_data/GREW/heatmap_sil
dataset_partition: ./datasets/GREW/GREW.json
num_workers: 1
data_in_use: [True, True] # heatmap, sil
@@ -10,7 +10,7 @@ data_cfg:
evaluator_cfg:
enable_float16: true
restore_ckpt_strict: true
restore_hint: 180000
restore_hint: /home/jdy/fanchao/OpenGait_230701/output/GREW/DeepGaitV2_P3D_GaitMap_MM_FeatureLevel/DeepGaitV2_P3D_GaitMap_B4C2_Sigma-8.0_Hot_False_Align-True_OpenGaitDA-True_LowLevel-AttentionFusion/checkpoints/DeepGaitV2_P3D_GaitMap_B4C2_Sigma-8.0_Hot_False_Align-True_OpenGaitDA-True_LowLevel-AttentionFusion-180000.pt
save_name: SkeletonGaitPP
eval_func: GREW_submission
sampler: