config.json

{
  "model": {
    "_target_": "detrpose.nn.DETR",
    "backbone": {
      "_target_": "detrpose.nn.HGNetv2",
      "freeze_at": -1,
      "freeze_norm": true,
      "freeze_stem_only": true,
      "name": "B4",
      "pretrained": false,
      "return_idx": [
        1,
        2,
        3
      ],
      "use_lab": false
    },
    "encoder": {
      "_target_": "detrpose.nn.DETRPoseHybridEncoder",
      "act": "silu",
      "depth_mult": 1.0,
      "dim_feedforward": 1024,
      "dropout": 0.0,
      "enc_act": "gelu",
      "eval_spatial_size": [
        640,
        640
      ],
      "expansion": 1.0,
      "feat_strides": [
        8,
        16,
        32
      ],
      "hidden_dim": 256,
      "in_channels": [
        512,
        1024,
        2048
      ],
      "num_heads": 8,
      "num_levels": 3,
      "temperatureH": 20,
      "temperatureW": 20
    },
    "transformer": {
      "_target_": "detrpose.nn.DETRPoseTransformer",
      "activation": "relu",
      "cls_no_bias": false,
      "dec_pred_class_embed_share": false,
      "dec_pred_pose_embed_share": false,
      "dim_feedforward": 1024,
      "dropout": 0.0,
      "eval_spatial_size": [
        640,
        640
      ],
      "feat_strides": [
        8,
        16,
        32
      ],
      "hidden_dim": 256,
      "learnable_tgt_init": true,
      "num_classes": 2,
      "num_heads": 8,
      "num_keypoints": 17,
      "num_layers": 6,
      "num_levels": 3,
      "num_points": 4,
      "num_queries": 60,
      "num_select": 60,
      "reg_max": 32,
      "reg_scale": 4,
      "return_intermediate_dec": true,
      "two_stage_bbox_embed_share": false,
      "two_stage_class_embed_share": false,
      "two_stage_type": "standard"
    }
  },
  "postprocessor": {
    "_target_": "detrpose.nn.PostProcessor",
    "num_keypoints": 17,
    "num_select": 60
  }
}