jln64.yaml

CUDNN:
  BENCHMARK: true
  DETERMINISTIC: false
  ENABLED: true

BACKBONE: 'resnet'
MODEL: 'faster_voxelpose'
DEVICE: 'cuda:0'
OUTPUT_DIR: 'output'
LOG_DIR: 'log'
WORKERS: 8
PRINT_FREQ: 100

DATASET:
  DATADIR: "data/Shelf"
  COLOR_RGB: True
  TRAIN_DATASET: 'synthetic'
  TRAIN_HEATMAP_SRC: 'gt'
  TEST_DATASET: 'shelf'
  TEST_HEATMAP_SRC: 'pred'
  CAMERA_NUM: 5
  ORI_IMAGE_SIZE: 
    - 1032
    - 776
  IMAGE_SIZE: 
    - 800
    - 608
  HEATMAP_SIZE: 
    - 200
    - 152
  NUM_JOINTS: 17
  ROOT_JOINT_ID:
    - 11
    - 12

SYNTHETIC:
  CAMERA_FILE: 'calibration_shelf.json'
  POSE_FILE: '../panoptic_training_pose.pkl'
  MAX_PEOPLE: 10
  NUM_DATA: 10000
  DATA_AUGMENTATION: true

NETWORK:
  PRETRAINED_BACKBONE: ""
  NUM_CHANNEL_JOINT_FEAT: 32
  NUM_CHANNEL_JOINT_HIDDEN: 64
  SIGMA: 3
  BETA: 100

TRAIN:
  BATCH_SIZE: 8
  SHUFFLE: true
  BEGIN_EPOCH: 0
  END_EPOCH: 50
  RESUME: false
  OPTIMIZER: adam
  LR: 0.0001
  LAMBDA_LOSS_2D: 1.0
  LAMBDA_LOSS_1D: 1.0
  LAMBDA_LOSS_BBOX: 0.1
  LAMBDA_LOSS_FUSED: 5.0
  VISUALIZATION: false
  VIS_TYPE: ['2d_planes']

TEST:
  MODEL_FILE: 'model_best.pth.tar'
  BATCH_SIZE: 16
  VISUALIZATION: false
  VIS_TYPE: ['2d_planes', 'image_with_poses']

CAPTURE_SPEC:
  SPACE_SIZE:
    - 8000.0
    - 8000.0
    - 2000.0
  SPACE_CENTER:
    - 450.0
    - -320.0
    - 800.0
  VOXELS_PER_AXIS:
    - 80
    - 80
    - 20
  MAX_PEOPLE: 10
  MIN_SCORE: 0.1
  
INDIVIDUAL_SPEC:
  SPACE_SIZE:
    - 2000.0
    - 2000.0
    - 2000.0
  VOXELS_PER_AXIS:
    - 64
    - 64
    - 64