# Inference configuration for the independently reproduced HRNet-R90JT checkpoint.
# Network dimensions match the configuration used by the MMInfant landmark pipeline.
# Training datasets, optimizer settings, and pretrained-initialization paths are omitted.
GPUS: (0, )
MODEL:
  NAME: hrnet
  NUM_JOINTS: 68
  INIT_WEIGHTS: false
  PRETRAINED: ''
  SIGMA: 1.0
  IMAGE_SIZE:
  - 256
  - 256
  HEATMAP_SIZE:
  - 64
  - 64
  EXTRA:
    FINAL_CONV_KERNEL: 1
    STAGE2:
      NUM_MODULES: 1
      NUM_BRANCHES: 2
      BLOCK: BASIC
      NUM_BLOCKS:
      - 4
      - 4
      NUM_CHANNELS:
      - 18
      - 36
      FUSE_METHOD: SUM
    STAGE3:
      NUM_MODULES: 4
      NUM_BRANCHES: 3
      BLOCK: BASIC
      NUM_BLOCKS:
      - 4
      - 4
      - 4
      NUM_CHANNELS:
      - 18
      - 36
      - 72
      FUSE_METHOD: SUM
    STAGE4:
      NUM_MODULES: 3
      NUM_BRANCHES: 4
      BLOCK: BASIC
      NUM_BLOCKS:
      - 4
      - 4
      - 4
      - 4
      NUM_CHANNELS:
      - 18
      - 36
      - 72
      - 144
      FUSE_METHOD: SUM
TEST:
  BATCH_SIZE_PER_GPU: 8