scannet_trainval.py 894 B

1234567891011121314151617
  1. from configs.data.base import cfg
  2. TRAIN_BASE_PATH = "data/scannet/index"
  3. cfg.DATASET.TRAINVAL_DATA_SOURCE = "ScanNet"
  4. cfg.DATASET.TRAIN_DATA_ROOT = "data/scannet/train"
  5. cfg.DATASET.TRAIN_NPZ_ROOT = f"{TRAIN_BASE_PATH}/scene_data/train"
  6. cfg.DATASET.TRAIN_LIST_PATH = f"{TRAIN_BASE_PATH}/scene_data/train_list/scannet_all.txt"
  7. cfg.DATASET.TRAIN_INTRINSIC_PATH = f"{TRAIN_BASE_PATH}/intrinsics.npz"
  8. TEST_BASE_PATH = "assets/scannet_test_1500"
  9. cfg.DATASET.TEST_DATA_SOURCE = "ScanNet"
  10. cfg.DATASET.VAL_DATA_ROOT = cfg.DATASET.TEST_DATA_ROOT = "data/scannet/test"
  11. cfg.DATASET.VAL_NPZ_ROOT = cfg.DATASET.TEST_NPZ_ROOT = TEST_BASE_PATH
  12. cfg.DATASET.VAL_LIST_PATH = cfg.DATASET.TEST_LIST_PATH = f"{TEST_BASE_PATH}/scannet_test.txt"
  13. cfg.DATASET.VAL_INTRINSIC_PATH = cfg.DATASET.TEST_INTRINSIC_PATH = f"{TEST_BASE_PATH}/intrinsics.npz"
  14. cfg.DATASET.MIN_OVERLAP_SCORE_TEST = 0.0 # for both test and val