config.json

{
  "model_type": "bucket-pose-estimation",
  "input_channels": 8,
  "input_size": [
    8,
    96,
    96
  ],
  "img_embedding_dim": 512,
  "num_keypoints": 6,
  "backbone": "resnet18",
  "random_init": true
}