crosstyan/OpenGait
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Align DRF skeleton preprocessing with upstream heatmap path

Browse Source
This commit is contained in:
crosstyan
2026-03-08 14:50:35 +08:00
parent bbb41e8dd9
commit 295d951206
10 changed files with 174 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files
configs/drf/drf_scoliosis1k.yaml
+3 -3
View File
@@ -1,6 +1,6 @@
data_cfg:
dataset_name: Scoliosis1K
dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl-118-paper
dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl-118-aligned
dataset_partition: ./datasets/Scoliosis1K/Scoliosis1K_118.json
num_workers: 1
remove_no_gallery: false
@@ -19,7 +19,7 @@ evaluator_cfg:
frames_all_limit: 720
metric: euc
transform:
- type: BaseSilTransform
- type: BaseSilCuttingTransform
- type: NoOperation
loss_cfg:
@@ -102,5 +102,5 @@ trainer_cfg:
sample_type: fixed_unordered
type: TripletSampler
transform:
- type: BaseSilTransform
- type: BaseSilCuttingTransform
- type: NoOperation
configs/drf/drf_scoliosis1k_eval_1gpu.yaml
+3 -3
View File
@@ -1,6 +1,6 @@
data_cfg:
dataset_name: Scoliosis1K
dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl-118-paper
dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl-118-aligned
dataset_partition: ./datasets/Scoliosis1K/Scoliosis1K_118.json
num_workers: 1
remove_no_gallery: false
@@ -19,7 +19,7 @@ evaluator_cfg:
frames_all_limit: 720
metric: euc
transform:
- type: BaseSilTransform
- type: BaseSilCuttingTransform
- type: NoOperation
loss_cfg:
@@ -102,5 +102,5 @@ trainer_cfg:
sample_type: fixed_unordered
type: TripletSampler
transform:
- type: BaseSilTransform
- type: BaseSilCuttingTransform
- type: NoOperation
configs/drf/drf_scoliosis1k_smoke.yaml
+3 -3
View File
@@ -1,6 +1,6 @@
data_cfg:
dataset_name: Scoliosis1K
dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl-118-paper
dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl-118-aligned
dataset_partition: ./datasets/Scoliosis1K/Scoliosis1K_118.json
num_workers: 1
remove_no_gallery: false
@@ -23,7 +23,7 @@ evaluator_cfg:
frames_all_limit: 720
metric: euc
transform:
- type: BaseSilTransform
- type: BaseSilCuttingTransform
- type: NoOperation
loss_cfg:
@@ -109,5 +109,5 @@ trainer_cfg:
sample_type: fixed_unordered
type: TripletSampler
transform:
- type: BaseSilTransform
- type: BaseSilCuttingTransform
- type: NoOperation
configs/drf/pretreatment_heatmap_drf.yaml
-1
View File
@@ -9,7 +9,6 @@ norm_args:
pose_format: coco
use_conf: ${padkeypoints_args.use_conf}
heatmap_image_height: 128
target_body_height: ${norm_args.heatmap_image_height}
heatmap_generator_args:
sigma: 8.0