| data: | |
| colormode: RGB | |
| inference: | |
| normalize_images: true | |
| top_down_crop: | |
| width: 288 | |
| height: 384 | |
| train: | |
| affine: | |
| p: 0.5 | |
| rotation: 30 | |
| scaling: | |
| - 1.0 | |
| - 1.0 | |
| translation: 0 | |
| collate: | |
| covering: false | |
| gaussian_noise: 12.75 | |
| hist_eq: false | |
| motion_blur: false | |
| normalize_images: true | |
| top_down_crop: | |
| width: 288 | |
| height: 384 | |
| detector: null | |
| device: auto | |
| metadata: | |
| project_path: null | |
| pose_config_path: rtmpose_x_body7_pytorch_config.yaml | |
| bodyparts: | |
| - nose | |
| - left_eye | |
| - right_eye | |
| - left_ear | |
| - right_ear | |
| - left_shoulder | |
| - right_shoulder | |
| - left_elbow | |
| - right_elbow | |
| - left_wrist | |
| - right_wrist | |
| - left_hip | |
| - right_hip | |
| - left_knee | |
| - right_knee | |
| - left_ankle | |
| - right_ankle | |
| unique_bodyparts: [] | |
| individuals: | |
| - idv0 | |
| - idv1 | |
| - idv2 | |
| - idv3 | |
| - idv4 | |
| - idv5 | |
| - idv6 | |
| - idv7 | |
| - idv8 | |
| - idv9 | |
| with_identity: false | |
| method: td | |
| model: | |
| backbone: | |
| type: CSPNeXt | |
| model_name: cspnext_p5 | |
| freeze_bn_stats: false | |
| freeze_bn_weights: false | |
| expand_ratio: 0.5 | |
| deepen_factor: 1.33 | |
| widen_factor: 1.25 | |
| channel_attention: true | |
| norm_layer: SyncBN | |
| activation_fn: SiLU | |
| backbone_output_channels: 1280 | |
| heads: | |
| bodypart: | |
| type: RTMCCHead | |
| weight_init: RTMPose | |
| target_generator: | |
| type: SimCCGenerator | |
| input_size: | |
| - 288 | |
| - 384 | |
| smoothing_type: gaussian | |
| sigma: | |
| - 6.0 | |
| - 6.93 | |
| simcc_split_ratio: 2.0 | |
| label_smooth_weight: 0.0 | |
| normalize: false | |
| criterion: | |
| x: | |
| type: KLDiscreteLoss | |
| use_target_weight: true | |
| beta: 10.0 | |
| label_softmax: true | |
| y: | |
| type: KLDiscreteLoss | |
| use_target_weight: true | |
| beta: 10.0 | |
| label_softmax: true | |
| predictor: | |
| type: SimCCPredictor | |
| simcc_split_ratio: 2.0 | |
| input_size: | |
| - 288 | |
| - 384 | |
| in_channels: 1280 | |
| out_channels: 17 | |
| in_featuremap_size: | |
| - 9 | |
| - 12 | |
| simcc_split_ratio: 2.0 | |
| final_layer_kernel_size: 7 | |
| gau_cfg: | |
| hidden_dims: 256 | |
| s: 128 | |
| expansion_factor: 2 | |
| dropout_rate: 0 | |
| drop_path: 0.0 | |
| act_fn: SiLU | |
| use_rel_bias: false | |
| pos_enc: false | |
| net_type: rtmpose_x | |
| runner: | |
| type: PoseTrainingRunner | |
| gpus: | |
| key_metric: test.mAP | |
| key_metric_asc: true | |
| eval_interval: 10 | |
| optimizer: | |
| type: AdamW | |
| params: | |
| lr: 0.0005 | |
| scheduler: | |
| type: SequentialLR | |
| params: | |
| schedulers: | |
| - type: ConstantLR | |
| params: | |
| factor: 0.001 | |
| total_iters: 5 | |
| - type: CosineAnnealingLR | |
| params: | |
| T_max: 250 | |
| eta_min: 1e-05 | |
| milestones: | |
| - 100 | |
| snapshots: | |
| max_snapshots: 5 | |
| save_epochs: 25 | |
| save_optimizer_state: false | |
| train_settings: | |
| batch_size: 1 | |
| dataloader_workers: 0 | |
| dataloader_pin_memory: false | |
| display_iters: 500 | |
| epochs: 200 | |
| seed: 42 | |