From 782bc098a62fb79c2a56734ee56f20f4e7a49fcb Mon Sep 17 00:00:00 2001 From: crosstyan Date: Tue, 10 Mar 2026 15:04:05 +0800 Subject: [PATCH] feat: add scoliosis body-only and head-lite experiment configs --- ...t_heatmap_drf_sigma15_joint8_bodyonly.yaml | 53 ++++++++ ...t_heatmap_drf_sigma15_joint8_headlite.yaml | 56 +++++++++ ...int8_bodyonly_plaince_bridge_1gpu_10k.yaml | 112 +++++++++++++++++ ...int8_bodyonly_plaince_bridge_2gpu_10k.yaml | 112 +++++++++++++++++ ...odyonly_plaince_bridge_eval_7000_1gpu.yaml | 59 +++++++++ ...8_bodyonly_weightedce_bridge_2gpu_10k.yaml | 116 ++++++++++++++++++ ...only_weightedce_bridge_eval_7000_1gpu.yaml | 113 +++++++++++++++++ ...int8_headlite_plaince_bridge_1gpu_10k.yaml | 112 +++++++++++++++++ ...18_sigma15_joint8_bodyonly_proxy_1gpu.yaml | 112 +++++++++++++++++ ...t8_bodyonly_weightedce_eval_1000_1gpu.yaml | 114 +++++++++++++++++ ...t8_bodyonly_weightedce_eval_2000_1gpu.yaml | 114 +++++++++++++++++ ...joint8_bodyonly_weightedce_proxy_1gpu.yaml | 116 ++++++++++++++++++ ...joint8_bodyonly_weightedce_proxy_2gpu.yaml | 116 ++++++++++++++++++ 13 files changed, 1305 insertions(+) create mode 100644 configs/drf/pretreatment_heatmap_drf_sigma15_joint8_bodyonly.yaml create mode 100644 configs/drf/pretreatment_heatmap_drf_sigma15_joint8_headlite.yaml create mode 100644 configs/sconet/sconet_scoliosis1k_skeleton_112_sigma15_joint8_bodyonly_plaince_bridge_1gpu_10k.yaml create mode 100644 configs/sconet/sconet_scoliosis1k_skeleton_112_sigma15_joint8_bodyonly_plaince_bridge_2gpu_10k.yaml create mode 100644 configs/sconet/sconet_scoliosis1k_skeleton_112_sigma15_joint8_bodyonly_plaince_bridge_eval_7000_1gpu.yaml create mode 100644 configs/sconet/sconet_scoliosis1k_skeleton_112_sigma15_joint8_bodyonly_weightedce_bridge_2gpu_10k.yaml create mode 100644 configs/sconet/sconet_scoliosis1k_skeleton_112_sigma15_joint8_bodyonly_weightedce_bridge_eval_7000_1gpu.yaml create mode 100644 configs/sconet/sconet_scoliosis1k_skeleton_112_sigma15_joint8_headlite_plaince_bridge_1gpu_10k.yaml create mode 100644 configs/sconet/sconet_scoliosis1k_skeleton_118_sigma15_joint8_bodyonly_proxy_1gpu.yaml create mode 100644 configs/sconet/sconet_scoliosis1k_skeleton_118_sigma15_joint8_bodyonly_weightedce_eval_1000_1gpu.yaml create mode 100644 configs/sconet/sconet_scoliosis1k_skeleton_118_sigma15_joint8_bodyonly_weightedce_eval_2000_1gpu.yaml create mode 100644 configs/sconet/sconet_scoliosis1k_skeleton_118_sigma15_joint8_bodyonly_weightedce_proxy_1gpu.yaml create mode 100644 configs/sconet/sconet_scoliosis1k_skeleton_118_sigma15_joint8_bodyonly_weightedce_proxy_2gpu.yaml diff --git a/configs/drf/pretreatment_heatmap_drf_sigma15_joint8_bodyonly.yaml b/configs/drf/pretreatment_heatmap_drf_sigma15_joint8_bodyonly.yaml new file mode 100644 index 0000000..ff16117 --- /dev/null +++ b/configs/drf/pretreatment_heatmap_drf_sigma15_joint8_bodyonly.yaml @@ -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} diff --git a/configs/drf/pretreatment_heatmap_drf_sigma15_joint8_headlite.yaml b/configs/drf/pretreatment_heatmap_drf_sigma15_joint8_headlite.yaml new file mode 100644 index 0000000..0faa750 --- /dev/null +++ b/configs/drf/pretreatment_heatmap_drf_sigma15_joint8_headlite.yaml @@ -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} diff --git a/configs/sconet/sconet_scoliosis1k_skeleton_112_sigma15_joint8_bodyonly_plaince_bridge_1gpu_10k.yaml b/configs/sconet/sconet_scoliosis1k_skeleton_112_sigma15_joint8_bodyonly_plaince_bridge_1gpu_10k.yaml new file mode 100644 index 0000000..eed7922 --- /dev/null +++ b/configs/sconet/sconet_scoliosis1k_skeleton_112_sigma15_joint8_bodyonly_plaince_bridge_1gpu_10k.yaml @@ -0,0 +1,112 @@ +data_cfg: + dataset_name: Scoliosis1K + dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl-118-sigma15-joint8-bodyonly + dataset_partition: ./datasets/Scoliosis1K/Scoliosis1K_112.json + data_in_use: + - true + - false + num_workers: 1 + remove_no_gallery: false + test_dataset_name: Scoliosis1K + test_seq_subset_size: 128 + test_seq_subset_seed: 112 + +evaluator_cfg: + enable_float16: true + restore_ckpt_strict: true + restore_hint: 10000 + save_name: ScoNet_skeleton_112_sigma15_joint8_bodyonly_plaince_bridge_1gpu_10k + eval_func: evaluate_scoliosis + sampler: + batch_shuffle: false + batch_size: 1 + sample_type: all_ordered + type: InferenceSampler + frames_all_limit: 720 + metric: euc + transform: + - type: BaseSilCuttingTransform + +loss_cfg: + - loss_term_weight: 1.0 + margin: 0.2 + type: TripletLoss + log_prefix: triplet + - loss_term_weight: 1.0 + scale: 16 + type: CrossEntropyLoss + log_prefix: softmax + log_accuracy: true + +model_cfg: + model: ScoNet + backbone_cfg: + type: ResNet9 + block: BasicBlock + in_channel: 2 + channels: + - 64 + - 128 + - 256 + - 512 + layers: + - 1 + - 1 + - 1 + - 1 + strides: + - 1 + - 2 + - 2 + - 1 + maxpool: false + SeparateFCs: + in_channels: 512 + out_channels: 256 + parts_num: 16 + SeparateBNNecks: + class_num: 3 + in_channels: 256 + parts_num: 16 + bin_num: + - 16 + +optimizer_cfg: + lr: 0.1 + momentum: 0.9 + solver: SGD + weight_decay: 0.0005 + +scheduler_cfg: + gamma: 0.1 + milestones: + - 5000 + - 7000 + - 9000 + scheduler: MultiStepLR + +trainer_cfg: + enable_float16: true + fix_BN: false + with_test: true + log_iter: 100 + restore_ckpt_strict: true + restore_hint: 0 + auto_resume_latest: true + resume_every_iter: 500 + resume_keep: 6 + eval_iter: 1000 + save_iter: 1000 + save_name: ScoNet_skeleton_112_sigma15_joint8_bodyonly_plaince_bridge_1gpu_10k + sync_BN: false + total_iter: 10000 + sampler: + batch_shuffle: true + batch_size: + - 8 + - 8 + frames_num_fixed: 30 + sample_type: fixed_unordered + type: TripletSampler + transform: + - type: BaseSilCuttingTransform diff --git a/configs/sconet/sconet_scoliosis1k_skeleton_112_sigma15_joint8_bodyonly_plaince_bridge_2gpu_10k.yaml b/configs/sconet/sconet_scoliosis1k_skeleton_112_sigma15_joint8_bodyonly_plaince_bridge_2gpu_10k.yaml new file mode 100644 index 0000000..82e8f2f --- /dev/null +++ b/configs/sconet/sconet_scoliosis1k_skeleton_112_sigma15_joint8_bodyonly_plaince_bridge_2gpu_10k.yaml @@ -0,0 +1,112 @@ +data_cfg: + dataset_name: Scoliosis1K + dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl-118-sigma15-joint8-bodyonly + dataset_partition: ./datasets/Scoliosis1K/Scoliosis1K_112.json + data_in_use: + - true + - false + num_workers: 1 + remove_no_gallery: false + test_dataset_name: Scoliosis1K + test_seq_subset_size: 128 + test_seq_subset_seed: 112 + +evaluator_cfg: + enable_float16: true + restore_ckpt_strict: true + restore_hint: 10000 + save_name: ScoNet_skeleton_112_sigma15_joint8_bodyonly_plaince_bridge_2gpu_10k + eval_func: evaluate_scoliosis + sampler: + batch_shuffle: false + batch_size: 2 + sample_type: all_ordered + type: InferenceSampler + frames_all_limit: 720 + metric: euc + transform: + - type: BaseSilCuttingTransform + +loss_cfg: + - loss_term_weight: 1.0 + margin: 0.2 + type: TripletLoss + log_prefix: triplet + - loss_term_weight: 1.0 + scale: 16 + type: CrossEntropyLoss + log_prefix: softmax + log_accuracy: true + +model_cfg: + model: ScoNet + backbone_cfg: + type: ResNet9 + block: BasicBlock + in_channel: 2 + channels: + - 64 + - 128 + - 256 + - 512 + layers: + - 1 + - 1 + - 1 + - 1 + strides: + - 1 + - 2 + - 2 + - 1 + maxpool: false + SeparateFCs: + in_channels: 512 + out_channels: 256 + parts_num: 16 + SeparateBNNecks: + class_num: 3 + in_channels: 256 + parts_num: 16 + bin_num: + - 16 + +optimizer_cfg: + lr: 0.1 + momentum: 0.9 + solver: SGD + weight_decay: 0.0005 + +scheduler_cfg: + gamma: 0.1 + milestones: + - 5000 + - 7000 + - 9000 + scheduler: MultiStepLR + +trainer_cfg: + enable_float16: true + fix_BN: false + with_test: true + log_iter: 100 + restore_ckpt_strict: true + restore_hint: 0 + auto_resume_latest: true + resume_every_iter: 500 + resume_keep: 6 + eval_iter: 1000 + save_iter: 1000 + save_name: ScoNet_skeleton_112_sigma15_joint8_bodyonly_plaince_bridge_2gpu_10k + sync_BN: true + total_iter: 10000 + sampler: + batch_shuffle: true + batch_size: + - 8 + - 8 + frames_num_fixed: 30 + sample_type: fixed_unordered + type: TripletSampler + transform: + - type: BaseSilCuttingTransform diff --git a/configs/sconet/sconet_scoliosis1k_skeleton_112_sigma15_joint8_bodyonly_plaince_bridge_eval_7000_1gpu.yaml b/configs/sconet/sconet_scoliosis1k_skeleton_112_sigma15_joint8_bodyonly_plaince_bridge_eval_7000_1gpu.yaml new file mode 100644 index 0000000..9e61cc6 --- /dev/null +++ b/configs/sconet/sconet_scoliosis1k_skeleton_112_sigma15_joint8_bodyonly_plaince_bridge_eval_7000_1gpu.yaml @@ -0,0 +1,59 @@ +data_cfg: + dataset_name: Scoliosis1K + dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl-118-sigma15-joint8-bodyonly + dataset_partition: ./datasets/Scoliosis1K/Scoliosis1K_112.json + data_in_use: + - true + - false + num_workers: 1 + remove_no_gallery: false + test_dataset_name: Scoliosis1K + +evaluator_cfg: + enable_float16: true + restore_ckpt_strict: true + restore_hint: /home/crosstyan/Code/OpenGait/output/Scoliosis1K/ScoNet/ScoNet_skeleton_112_sigma15_joint8_bodyonly_plaince_bridge_1gpu_10k/checkpoints/ScoNet_skeleton_112_sigma15_joint8_bodyonly_plaince_bridge_1gpu_10k-07000.pt + save_name: ScoNet_skeleton_112_sigma15_joint8_bodyonly_plaince_bridge_eval_7000_1gpu + eval_func: evaluate_scoliosis + sampler: + batch_shuffle: false + batch_size: 1 + sample_type: all_ordered + type: InferenceSampler + frames_all_limit: 720 + metric: euc + transform: + - type: BaseSilCuttingTransform + +model_cfg: + model: ScoNet + backbone_cfg: + type: ResNet9 + block: BasicBlock + in_channel: 2 + channels: + - 64 + - 128 + - 256 + - 512 + layers: + - 1 + - 1 + - 1 + - 1 + strides: + - 1 + - 2 + - 2 + - 1 + maxpool: false + SeparateFCs: + in_channels: 512 + out_channels: 256 + parts_num: 16 + SeparateBNNecks: + class_num: 3 + in_channels: 256 + parts_num: 16 + bin_num: + - 16 diff --git a/configs/sconet/sconet_scoliosis1k_skeleton_112_sigma15_joint8_bodyonly_weightedce_bridge_2gpu_10k.yaml b/configs/sconet/sconet_scoliosis1k_skeleton_112_sigma15_joint8_bodyonly_weightedce_bridge_2gpu_10k.yaml new file mode 100644 index 0000000..a620666 --- /dev/null +++ b/configs/sconet/sconet_scoliosis1k_skeleton_112_sigma15_joint8_bodyonly_weightedce_bridge_2gpu_10k.yaml @@ -0,0 +1,116 @@ +data_cfg: + dataset_name: Scoliosis1K + dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl-118-sigma15-joint8-bodyonly + dataset_partition: ./datasets/Scoliosis1K/Scoliosis1K_112.json + data_in_use: + - true + - false + num_workers: 1 + remove_no_gallery: false + test_dataset_name: Scoliosis1K + test_seq_subset_size: 128 + test_seq_subset_seed: 112 + +evaluator_cfg: + enable_float16: true + restore_ckpt_strict: true + restore_hint: 10000 + save_name: ScoNet_skeleton_112_sigma15_joint8_bodyonly_weightedce_bridge_2gpu_10k + eval_func: evaluate_scoliosis + sampler: + batch_shuffle: false + batch_size: 2 + sample_type: all_ordered + type: InferenceSampler + frames_all_limit: 720 + metric: euc + transform: + - type: BaseSilCuttingTransform + +loss_cfg: + - loss_term_weight: 1.0 + margin: 0.2 + type: TripletLoss + log_prefix: triplet + - loss_term_weight: 1.0 + scale: 16 + type: CrossEntropyLoss + log_prefix: softmax + log_accuracy: true + class_weight: + - 1.0 + - 4.0 + - 4.0 + +model_cfg: + model: ScoNet + backbone_cfg: + type: ResNet9 + block: BasicBlock + in_channel: 2 + channels: + - 64 + - 128 + - 256 + - 512 + layers: + - 1 + - 1 + - 1 + - 1 + strides: + - 1 + - 2 + - 2 + - 1 + maxpool: false + SeparateFCs: + in_channels: 512 + out_channels: 256 + parts_num: 16 + SeparateBNNecks: + class_num: 3 + in_channels: 256 + parts_num: 16 + bin_num: + - 16 + +optimizer_cfg: + lr: 0.1 + momentum: 0.9 + solver: SGD + weight_decay: 0.0005 + +scheduler_cfg: + gamma: 0.1 + milestones: + - 5000 + - 7000 + - 9000 + scheduler: MultiStepLR + +trainer_cfg: + enable_float16: true + fix_BN: false + with_test: true + log_iter: 100 + restore_ckpt_strict: true + restore_hint: 0 + auto_resume_latest: true + resume_every_iter: 500 + resume_keep: 6 + eval_iter: 1000 + save_iter: 1000 + save_name: ScoNet_skeleton_112_sigma15_joint8_bodyonly_weightedce_bridge_2gpu_10k + sync_BN: true + total_iter: 10000 + sampler: + batch_shuffle: true + batch_size: + - 8 + - 8 + frames_num_fixed: 30 + sample_type: fixed_unordered + type: TripletSampler + transform: + - type: BaseSilCuttingTransform diff --git a/configs/sconet/sconet_scoliosis1k_skeleton_112_sigma15_joint8_bodyonly_weightedce_bridge_eval_7000_1gpu.yaml b/configs/sconet/sconet_scoliosis1k_skeleton_112_sigma15_joint8_bodyonly_weightedce_bridge_eval_7000_1gpu.yaml new file mode 100644 index 0000000..d9b8305 --- /dev/null +++ b/configs/sconet/sconet_scoliosis1k_skeleton_112_sigma15_joint8_bodyonly_weightedce_bridge_eval_7000_1gpu.yaml @@ -0,0 +1,113 @@ +data_cfg: + dataset_name: Scoliosis1K + dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl-118-sigma15-joint8-bodyonly + dataset_partition: ./datasets/Scoliosis1K/Scoliosis1K_112.json + data_in_use: + - true + - false + num_workers: 1 + remove_no_gallery: false + test_dataset_name: Scoliosis1K + +evaluator_cfg: + enable_float16: true + restore_ckpt_strict: true + restore_hint: /home/crosstyan/Code/OpenGait/output/Scoliosis1K/ScoNet/ScoNet_skeleton_112_sigma15_joint8_bodyonly_weightedce_bridge_2gpu_10k/checkpoints/ScoNet_skeleton_112_sigma15_joint8_bodyonly_weightedce_bridge_2gpu_10k-07000.pt + save_name: ScoNet_skeleton_112_sigma15_joint8_bodyonly_weightedce_bridge_eval_7000_1gpu + eval_func: evaluate_scoliosis + sampler: + batch_shuffle: false + batch_size: 1 + sample_type: all_ordered + type: InferenceSampler + frames_all_limit: 720 + metric: euc + transform: + - type: BaseSilCuttingTransform + +loss_cfg: + - loss_term_weight: 1.0 + margin: 0.2 + type: TripletLoss + log_prefix: triplet + - loss_term_weight: 1.0 + scale: 16 + type: CrossEntropyLoss + log_prefix: softmax + log_accuracy: true + class_weight: + - 1.0 + - 4.0 + - 4.0 + +model_cfg: + model: ScoNet + backbone_cfg: + type: ResNet9 + block: BasicBlock + in_channel: 2 + channels: + - 64 + - 128 + - 256 + - 512 + layers: + - 1 + - 1 + - 1 + - 1 + strides: + - 1 + - 2 + - 2 + - 1 + maxpool: false + SeparateFCs: + in_channels: 512 + out_channels: 256 + parts_num: 16 + SeparateBNNecks: + class_num: 3 + in_channels: 256 + parts_num: 16 + bin_num: + - 16 + +optimizer_cfg: + lr: 0.1 + momentum: 0.9 + solver: SGD + weight_decay: 0.0005 + +scheduler_cfg: + gamma: 0.1 + milestones: + - 5000 + - 7000 + - 9000 + scheduler: MultiStepLR + +trainer_cfg: + enable_float16: true + fix_BN: false + with_test: false + log_iter: 100 + restore_ckpt_strict: true + restore_hint: 0 + auto_resume_latest: true + resume_every_iter: 500 + resume_keep: 6 + save_iter: 1000 + save_name: ScoNet_skeleton_112_sigma15_joint8_bodyonly_weightedce_bridge_eval_7000_1gpu + sync_BN: false + total_iter: 10000 + sampler: + batch_shuffle: true + batch_size: + - 8 + - 8 + frames_num_fixed: 30 + sample_type: fixed_unordered + type: TripletSampler + transform: + - type: BaseSilCuttingTransform diff --git a/configs/sconet/sconet_scoliosis1k_skeleton_112_sigma15_joint8_headlite_plaince_bridge_1gpu_10k.yaml b/configs/sconet/sconet_scoliosis1k_skeleton_112_sigma15_joint8_headlite_plaince_bridge_1gpu_10k.yaml new file mode 100644 index 0000000..a2b2b06 --- /dev/null +++ b/configs/sconet/sconet_scoliosis1k_skeleton_112_sigma15_joint8_headlite_plaince_bridge_1gpu_10k.yaml @@ -0,0 +1,112 @@ +data_cfg: + dataset_name: Scoliosis1K + dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl-112-sigma15-joint8-headlite + dataset_partition: ./datasets/Scoliosis1K/Scoliosis1K_112.json + data_in_use: + - true + - false + num_workers: 1 + remove_no_gallery: false + test_dataset_name: Scoliosis1K + test_seq_subset_size: 128 + test_seq_subset_seed: 112 + +evaluator_cfg: + enable_float16: true + restore_ckpt_strict: true + restore_hint: 10000 + save_name: ScoNet_skeleton_112_sigma15_joint8_headlite_plaince_bridge_1gpu_10k + eval_func: evaluate_scoliosis + sampler: + batch_shuffle: false + batch_size: 1 + sample_type: all_ordered + type: InferenceSampler + frames_all_limit: 720 + metric: euc + transform: + - type: BaseSilCuttingTransform + +loss_cfg: + - loss_term_weight: 1.0 + margin: 0.2 + type: TripletLoss + log_prefix: triplet + - loss_term_weight: 1.0 + scale: 16 + type: CrossEntropyLoss + log_prefix: softmax + log_accuracy: true + +model_cfg: + model: ScoNet + backbone_cfg: + type: ResNet9 + block: BasicBlock + in_channel: 2 + channels: + - 64 + - 128 + - 256 + - 512 + layers: + - 1 + - 1 + - 1 + - 1 + strides: + - 1 + - 2 + - 2 + - 1 + maxpool: false + SeparateFCs: + in_channels: 512 + out_channels: 256 + parts_num: 16 + SeparateBNNecks: + class_num: 3 + in_channels: 256 + parts_num: 16 + bin_num: + - 16 + +optimizer_cfg: + lr: 0.1 + momentum: 0.9 + solver: SGD + weight_decay: 0.0005 + +scheduler_cfg: + gamma: 0.1 + milestones: + - 5000 + - 7000 + - 9000 + scheduler: MultiStepLR + +trainer_cfg: + enable_float16: true + fix_BN: false + with_test: true + log_iter: 100 + restore_ckpt_strict: true + restore_hint: 0 + auto_resume_latest: true + resume_every_iter: 500 + resume_keep: 6 + eval_iter: 1000 + save_iter: 1000 + save_name: ScoNet_skeleton_112_sigma15_joint8_headlite_plaince_bridge_1gpu_10k + sync_BN: false + total_iter: 10000 + sampler: + batch_shuffle: true + batch_size: + - 8 + - 8 + frames_num_fixed: 30 + sample_type: fixed_unordered + type: TripletSampler + transform: + - type: BaseSilCuttingTransform diff --git a/configs/sconet/sconet_scoliosis1k_skeleton_118_sigma15_joint8_bodyonly_proxy_1gpu.yaml b/configs/sconet/sconet_scoliosis1k_skeleton_118_sigma15_joint8_bodyonly_proxy_1gpu.yaml new file mode 100644 index 0000000..2c22154 --- /dev/null +++ b/configs/sconet/sconet_scoliosis1k_skeleton_118_sigma15_joint8_bodyonly_proxy_1gpu.yaml @@ -0,0 +1,112 @@ +data_cfg: + dataset_name: Scoliosis1K + dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl-118-sigma15-joint8-bodyonly + dataset_partition: ./datasets/Scoliosis1K/Scoliosis1K_118.json + data_in_use: + - true + - false + num_workers: 1 + remove_no_gallery: false + test_dataset_name: Scoliosis1K + test_seq_subset_size: 128 + test_seq_subset_seed: 118 + +evaluator_cfg: + enable_float16: true + restore_ckpt_strict: true + restore_hint: 2000 + save_name: ScoNet_skeleton_118_sigma15_joint8_bodyonly_proxy_1gpu + eval_func: evaluate_scoliosis + sampler: + batch_shuffle: false + batch_size: 1 + sample_type: all_ordered + type: InferenceSampler + frames_all_limit: 720 + metric: euc + transform: + - type: BaseSilCuttingTransform + +loss_cfg: + - loss_term_weight: 1.0 + margin: 0.2 + type: TripletLoss + log_prefix: triplet + - loss_term_weight: 1.0 + scale: 16 + type: CrossEntropyLoss + log_prefix: softmax + log_accuracy: true + +model_cfg: + model: ScoNet + backbone_cfg: + type: ResNet9 + block: BasicBlock + in_channel: 2 + channels: + - 64 + - 128 + - 256 + - 512 + layers: + - 1 + - 1 + - 1 + - 1 + strides: + - 1 + - 2 + - 2 + - 1 + maxpool: false + SeparateFCs: + in_channels: 512 + out_channels: 256 + parts_num: 16 + SeparateBNNecks: + class_num: 3 + in_channels: 256 + parts_num: 16 + bin_num: + - 16 + +optimizer_cfg: + lr: 0.1 + momentum: 0.9 + solver: SGD + weight_decay: 0.0005 + +scheduler_cfg: + gamma: 0.1 + milestones: + - 1000 + - 1500 + - 1800 + scheduler: MultiStepLR + +trainer_cfg: + enable_float16: true + fix_BN: false + with_test: true + log_iter: 100 + restore_ckpt_strict: true + restore_hint: 0 + auto_resume_latest: true + resume_every_iter: 500 + resume_keep: 3 + eval_iter: 500 + save_iter: 500 + save_name: ScoNet_skeleton_118_sigma15_joint8_bodyonly_proxy_1gpu + sync_BN: true + total_iter: 2000 + sampler: + batch_shuffle: true + batch_size: + - 8 + - 8 + frames_num_fixed: 30 + sample_type: fixed_unordered + type: TripletSampler + transform: + - type: BaseSilCuttingTransform diff --git a/configs/sconet/sconet_scoliosis1k_skeleton_118_sigma15_joint8_bodyonly_weightedce_eval_1000_1gpu.yaml b/configs/sconet/sconet_scoliosis1k_skeleton_118_sigma15_joint8_bodyonly_weightedce_eval_1000_1gpu.yaml new file mode 100644 index 0000000..49366d6 --- /dev/null +++ b/configs/sconet/sconet_scoliosis1k_skeleton_118_sigma15_joint8_bodyonly_weightedce_eval_1000_1gpu.yaml @@ -0,0 +1,114 @@ +data_cfg: + dataset_name: Scoliosis1K + dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl-118-sigma15-joint8-bodyonly + dataset_partition: ./datasets/Scoliosis1K/Scoliosis1K_118.json + data_in_use: + - true + - false + num_workers: 1 + remove_no_gallery: false + test_dataset_name: Scoliosis1K + +evaluator_cfg: + enable_float16: true + restore_ckpt_strict: true + restore_hint: /home/crosstyan/Code/OpenGait/output/Scoliosis1K/ScoNet/ScoNet_skeleton_118_sigma15_joint8_bodyonly_weightedce_proxy_2gpu/checkpoints/resume/ScoNet_skeleton_118_sigma15_joint8_bodyonly_weightedce_proxy_2gpu-resume-01000.pt + save_name: ScoNet_skeleton_118_sigma15_joint8_bodyonly_weightedce_eval_1000_1gpu + eval_func: evaluate_scoliosis + sampler: + batch_shuffle: false + batch_size: 1 + sample_type: all_ordered + type: InferenceSampler + frames_all_limit: 720 + metric: euc + transform: + - type: BaseSilCuttingTransform + +loss_cfg: + - loss_term_weight: 1.0 + margin: 0.2 + type: TripletLoss + log_prefix: triplet + - loss_term_weight: 1.0 + scale: 16 + type: CrossEntropyLoss + log_prefix: softmax + log_accuracy: true + class_weight: + - 1.0 + - 4.0 + - 4.0 + +model_cfg: + model: ScoNet + backbone_cfg: + type: ResNet9 + block: BasicBlock + in_channel: 2 + channels: + - 64 + - 128 + - 256 + - 512 + layers: + - 1 + - 1 + - 1 + - 1 + strides: + - 1 + - 2 + - 2 + - 1 + maxpool: false + SeparateFCs: + in_channels: 512 + out_channels: 256 + parts_num: 16 + SeparateBNNecks: + class_num: 3 + in_channels: 256 + parts_num: 16 + bin_num: + - 16 + +optimizer_cfg: + lr: 0.1 + momentum: 0.9 + solver: SGD + weight_decay: 0.0005 + +scheduler_cfg: + gamma: 0.1 + milestones: + - 1000 + - 1500 + - 1800 + scheduler: MultiStepLR + +trainer_cfg: + enable_float16: true + fix_BN: false + with_test: true + log_iter: 100 + restore_ckpt_strict: true + restore_hint: 0 + auto_resume_latest: true + resume_every_iter: 500 + resume_keep: 3 + eval_iter: 500 + save_iter: 2000 + save_name: ScoNet_skeleton_118_sigma15_joint8_bodyonly_weightedce_eval_1000_1gpu + sync_BN: true + total_iter: 2000 + sampler: + batch_shuffle: true + batch_size: + - 8 + - 8 + frames_num_fixed: 30 + sample_type: fixed_unordered + type: TripletSampler + transform: + - type: BaseSilCuttingTransform diff --git a/configs/sconet/sconet_scoliosis1k_skeleton_118_sigma15_joint8_bodyonly_weightedce_eval_2000_1gpu.yaml b/configs/sconet/sconet_scoliosis1k_skeleton_118_sigma15_joint8_bodyonly_weightedce_eval_2000_1gpu.yaml new file mode 100644 index 0000000..7c0d60a --- /dev/null +++ b/configs/sconet/sconet_scoliosis1k_skeleton_118_sigma15_joint8_bodyonly_weightedce_eval_2000_1gpu.yaml @@ -0,0 +1,114 @@ +data_cfg: + dataset_name: Scoliosis1K + dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl-118-sigma15-joint8-bodyonly + dataset_partition: ./datasets/Scoliosis1K/Scoliosis1K_118.json + data_in_use: + - true + - false + num_workers: 1 + remove_no_gallery: false + test_dataset_name: Scoliosis1K + +evaluator_cfg: + enable_float16: true + restore_ckpt_strict: true + restore_hint: /home/crosstyan/Code/OpenGait/output/Scoliosis1K/ScoNet/ScoNet_skeleton_118_sigma15_joint8_bodyonly_weightedce_proxy_2gpu/checkpoints/ScoNet_skeleton_118_sigma15_joint8_bodyonly_weightedce_proxy_2gpu-02000.pt + save_name: ScoNet_skeleton_118_sigma15_joint8_bodyonly_weightedce_eval_2000_1gpu + eval_func: evaluate_scoliosis + sampler: + batch_shuffle: false + batch_size: 1 + sample_type: all_ordered + type: InferenceSampler + frames_all_limit: 720 + metric: euc + transform: + - type: BaseSilCuttingTransform + +loss_cfg: + - loss_term_weight: 1.0 + margin: 0.2 + type: TripletLoss + log_prefix: triplet + - loss_term_weight: 1.0 + scale: 16 + type: CrossEntropyLoss + log_prefix: softmax + log_accuracy: true + class_weight: + - 1.0 + - 4.0 + - 4.0 + +model_cfg: + model: ScoNet + backbone_cfg: + type: ResNet9 + block: BasicBlock + in_channel: 2 + channels: + - 64 + - 128 + - 256 + - 512 + layers: + - 1 + - 1 + - 1 + - 1 + strides: + - 1 + - 2 + - 2 + - 1 + maxpool: false + SeparateFCs: + in_channels: 512 + out_channels: 256 + parts_num: 16 + SeparateBNNecks: + class_num: 3 + in_channels: 256 + parts_num: 16 + bin_num: + - 16 + +optimizer_cfg: + lr: 0.1 + momentum: 0.9 + solver: SGD + weight_decay: 0.0005 + +scheduler_cfg: + gamma: 0.1 + milestones: + - 1000 + - 1500 + - 1800 + scheduler: MultiStepLR + +trainer_cfg: + enable_float16: true + fix_BN: false + with_test: true + log_iter: 100 + restore_ckpt_strict: true + restore_hint: 0 + auto_resume_latest: true + resume_every_iter: 500 + resume_keep: 3 + eval_iter: 500 + save_iter: 2000 + save_name: ScoNet_skeleton_118_sigma15_joint8_bodyonly_weightedce_eval_2000_1gpu + sync_BN: true + total_iter: 2000 + sampler: + batch_shuffle: true + batch_size: + - 8 + - 8 + frames_num_fixed: 30 + sample_type: fixed_unordered + type: TripletSampler + transform: + - type: BaseSilCuttingTransform diff --git a/configs/sconet/sconet_scoliosis1k_skeleton_118_sigma15_joint8_bodyonly_weightedce_proxy_1gpu.yaml b/configs/sconet/sconet_scoliosis1k_skeleton_118_sigma15_joint8_bodyonly_weightedce_proxy_1gpu.yaml new file mode 100644 index 0000000..2b8a83b --- /dev/null +++ b/configs/sconet/sconet_scoliosis1k_skeleton_118_sigma15_joint8_bodyonly_weightedce_proxy_1gpu.yaml @@ -0,0 +1,116 @@ +data_cfg: + dataset_name: Scoliosis1K + dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl-118-sigma15-joint8-bodyonly + dataset_partition: ./datasets/Scoliosis1K/Scoliosis1K_118.json + data_in_use: + - true + - false + num_workers: 1 + remove_no_gallery: false + test_dataset_name: Scoliosis1K + test_seq_subset_size: 128 + test_seq_subset_seed: 118 + +evaluator_cfg: + enable_float16: true + restore_ckpt_strict: true + restore_hint: 2000 + save_name: ScoNet_skeleton_118_sigma15_joint8_bodyonly_weightedce_proxy_1gpu + eval_func: evaluate_scoliosis + sampler: + batch_shuffle: false + batch_size: 1 + sample_type: all_ordered + type: InferenceSampler + frames_all_limit: 720 + metric: euc + transform: + - type: BaseSilCuttingTransform + +loss_cfg: + - loss_term_weight: 1.0 + margin: 0.2 + type: TripletLoss + log_prefix: triplet + - loss_term_weight: 1.0 + scale: 16 + type: CrossEntropyLoss + log_prefix: softmax + log_accuracy: true + class_weight: + - 1.0 + - 4.0 + - 4.0 + +model_cfg: + model: ScoNet + backbone_cfg: + type: ResNet9 + block: BasicBlock + in_channel: 2 + channels: + - 64 + - 128 + - 256 + - 512 + layers: + - 1 + - 1 + - 1 + - 1 + strides: + - 1 + - 2 + - 2 + - 1 + maxpool: false + SeparateFCs: + in_channels: 512 + out_channels: 256 + parts_num: 16 + SeparateBNNecks: + class_num: 3 + in_channels: 256 + parts_num: 16 + bin_num: + - 16 + +optimizer_cfg: + lr: 0.1 + momentum: 0.9 + solver: SGD + weight_decay: 0.0005 + +scheduler_cfg: + gamma: 0.1 + milestones: + - 1000 + - 1500 + - 1800 + scheduler: MultiStepLR + +trainer_cfg: + enable_float16: true + fix_BN: false + with_test: true + log_iter: 100 + restore_ckpt_strict: true + restore_hint: 0 + auto_resume_latest: true + resume_every_iter: 500 + resume_keep: 3 + eval_iter: 500 + save_iter: 500 + save_name: ScoNet_skeleton_118_sigma15_joint8_bodyonly_weightedce_proxy_1gpu + sync_BN: true + total_iter: 2000 + sampler: + batch_shuffle: true + batch_size: + - 8 + - 8 + frames_num_fixed: 30 + sample_type: fixed_unordered + type: TripletSampler + transform: + - type: BaseSilCuttingTransform diff --git a/configs/sconet/sconet_scoliosis1k_skeleton_118_sigma15_joint8_bodyonly_weightedce_proxy_2gpu.yaml b/configs/sconet/sconet_scoliosis1k_skeleton_118_sigma15_joint8_bodyonly_weightedce_proxy_2gpu.yaml new file mode 100644 index 0000000..33ee2d8 --- /dev/null +++ b/configs/sconet/sconet_scoliosis1k_skeleton_118_sigma15_joint8_bodyonly_weightedce_proxy_2gpu.yaml @@ -0,0 +1,116 @@ +data_cfg: + dataset_name: Scoliosis1K + dataset_root: /mnt/public/data/Scoliosis1K/Scoliosis1K-drf-pkl-118-sigma15-joint8-bodyonly + dataset_partition: ./datasets/Scoliosis1K/Scoliosis1K_118.json + data_in_use: + - true + - false + num_workers: 1 + remove_no_gallery: false + test_dataset_name: Scoliosis1K + test_seq_subset_size: 128 + test_seq_subset_seed: 118 + +evaluator_cfg: + enable_float16: true + restore_ckpt_strict: true + restore_hint: 2000 + save_name: ScoNet_skeleton_118_sigma15_joint8_bodyonly_weightedce_proxy_2gpu + eval_func: evaluate_scoliosis + sampler: + batch_shuffle: false + batch_size: 2 + sample_type: all_ordered + type: InferenceSampler + frames_all_limit: 720 + metric: euc + transform: + - type: BaseSilCuttingTransform + +loss_cfg: + - loss_term_weight: 1.0 + margin: 0.2 + type: TripletLoss + log_prefix: triplet + - loss_term_weight: 1.0 + scale: 16 + type: CrossEntropyLoss + log_prefix: softmax + log_accuracy: true + class_weight: + - 1.0 + - 4.0 + - 4.0 + +model_cfg: + model: ScoNet + backbone_cfg: + type: ResNet9 + block: BasicBlock + in_channel: 2 + channels: + - 64 + - 128 + - 256 + - 512 + layers: + - 1 + - 1 + - 1 + - 1 + strides: + - 1 + - 2 + - 2 + - 1 + maxpool: false + SeparateFCs: + in_channels: 512 + out_channels: 256 + parts_num: 16 + SeparateBNNecks: + class_num: 3 + in_channels: 256 + parts_num: 16 + bin_num: + - 16 + +optimizer_cfg: + lr: 0.1 + momentum: 0.9 + solver: SGD + weight_decay: 0.0005 + +scheduler_cfg: + gamma: 0.1 + milestones: + - 1000 + - 1500 + - 1800 + scheduler: MultiStepLR + +trainer_cfg: + enable_float16: true + fix_BN: false + with_test: true + log_iter: 100 + restore_ckpt_strict: true + restore_hint: 0 + auto_resume_latest: true + resume_every_iter: 500 + resume_keep: 3 + eval_iter: 500 + save_iter: 500 + save_name: ScoNet_skeleton_118_sigma15_joint8_bodyonly_weightedce_proxy_2gpu + sync_BN: true + total_iter: 2000 + sampler: + batch_shuffle: true + batch_size: + - 8 + - 8 + frames_num_fixed: 30 + sample_type: fixed_unordered + type: TripletSampler + transform: + - type: BaseSilCuttingTransform