demo.yaml

DATASET: 'demo'
BATCH_SIZE: 1
SAVE_SCENE_MESH: True
SAVE_FRAGMENTS: False
LOGDIR: './checkpoints/release'
SAVE_FREQ: 1
MODE: 'test'

TRAIN:
  PATH: 'data/scannet'
  EPOCHS: 991
  N_VIEWS: 9
  LR: 1e-3
  N_WORKERS: 8
  LREPOCHS: '200,400,800:2'

TEST:
  PATH: '/work/vig/Datasets/PlanarRecon/demo_data/2022-05-09T17-21-06'
  N_VIEWS: 9
  N_WORKERS: 4  

MODEL:
  N_LAYER: 3
  N_VOX: [96, 96, 96]
  VOXEL_SIZE: 0.04
  TRAIN_NUM_SAMPLE: [4096, 16384, 65536]
  TEST_NUM_SAMPLE: [4096, 16384, 65536]

  BACKBONE2D:
    ARC: 'fpn-mnas-1'

  FUSION:
    FUSION_ON: True
    HIDDEN_DIM: 64


  LW: [1.0, 0.8, 0.64, 2.0]

  THRESHOLDS: [0, 0, 0]
  POS_WEIGHT: 1.5

  TRACKING: True

  NORMAL_ANCHOR_PATH: './data/scannet/planes_9/normal_anchors.npy'