File size: 477 Bytes
6979012 | 1 2 3 4 5 6 7 8 9 10 | from pathlib import Path
PACKAGE_ROOT = Path(__file__).resolve().parent
DATA_ROOT = PACKAGE_ROOT / "distillation" / "data"
DEFAULT_CONFIG_PATH = PACKAGE_ROOT / "distillation" / "base_config.yaml"
DEFAULT_WEIGHTS_PATH = PACKAGE_ROOT / "weights" / "superpoint_v6_from_tf.pth"
DEFAULT_DATASET_NAME = "indoor_forward_3_snapdragon_with_gt"
DEFAULT_IMAGE_DIR = DATA_ROOT / "datasets" / DEFAULT_DATASET_NAME / "img"
DEFAULT_TRAINING_IDS = DATA_ROOT / "gt_id_file_train_base_250.txt"
|