feat: add scoliosis body-only and head-lite experiment configs

2026-03-10 15:04:05 +08:00
parent 44e62ae3ae
commit 782bc098a6
13 changed files with 1305 additions and 0 deletions
@@ -0,0 +1,53 @@
+coco18tococo17_args:
+  transfer_to_coco17: False
+
+padkeypoints_args:
+  pad_method: knn
+  use_conf: True
+
+norm_args:
+  pose_format: coco
+  use_conf: ${padkeypoints_args.use_conf}
+  heatmap_image_height: 128
+
+heatmap_generator_args:
+  sigma: 1.5
+  use_score: ${padkeypoints_args.use_conf}
+  img_h: ${norm_args.heatmap_image_height}
+  img_w: ${norm_args.heatmap_image_height}
+  with_limb: null
+  with_kp: null
+  joint_indices:
+    - 5
+    - 6
+    - 7
+    - 8
+    - 9
+    - 10
+    - 11
+    - 12
+    - 13
+    - 14
+    - 15
+    - 16
+  skeletons:
+    - [5, 7]
+    - [7, 9]
+    - [6, 8]
+    - [8, 10]
+    - [5, 11]
+    - [6, 12]
+    - [11, 13]
+    - [13, 15]
+    - [12, 14]
+    - [14, 16]
+    - [11, 12]
+
+sigma_limb: 1.5
+sigma_joint: 8.0
+
+align_args:
+  align: True
+  final_img_size: 64
+  offset: 0
+  heatmap_image_size: ${norm_args.heatmap_image_height}
@@ -0,0 +1,56 @@
+coco18tococo17_args:
+  transfer_to_coco17: False
+
+padkeypoints_args:
+  pad_method: knn
+  use_conf: True
+
+norm_args:
+  pose_format: coco
+  use_conf: ${padkeypoints_args.use_conf}
+  heatmap_image_height: 128
+
+heatmap_generator_args:
+  sigma: 1.5
+  use_score: ${padkeypoints_args.use_conf}
+  img_h: ${norm_args.heatmap_image_height}
+  img_w: ${norm_args.heatmap_image_height}
+  with_limb: null
+  with_kp: null
+  joint_indices:
+    - 0
+    - 5
+    - 6
+    - 7
+    - 8
+    - 9
+    - 10
+    - 11
+    - 12
+    - 13
+    - 14
+    - 15
+    - 16
+  skeletons:
+    - [0, 5]
+    - [0, 6]
+    - [5, 7]
+    - [7, 9]
+    - [6, 8]
+    - [8, 10]
+    - [5, 11]
+    - [6, 12]
+    - [11, 13]
+    - [13, 15]
+    - [12, 14]
+    - [14, 16]
+    - [11, 12]
+
+sigma_limb: 1.5
+sigma_joint: 8.0
+
+align_args:
+  align: True
+  final_img_size: 64
+  offset: 0
+  heatmap_image_size: ${norm_args.heatmap_image_height}