File size: 3,068 Bytes
25986db | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 | import os
class EnvironmentSettings:
def __init__(self):
project_dir = os.environ.get(
'ATCTRACK_VLM_PROJECT_DIR',
os.path.abspath(os.path.join(os.path.dirname(__file__), '../../..')),
)
dataset_root = os.environ.get(
'DATASET_ROOT',
os.path.abspath(os.path.join(project_dir, '../Dataset')),
)
self.workspace_dir = os.environ.get('WORKSPACE_DIR', project_dir)
self.tensorboard_dir = os.environ.get('TENSORBOARD_DIR', os.path.join(project_dir, 'tensorboard'))
self.pretrained_networks = os.environ.get(
'PRETRAINED_NETWORKS_DIR',
os.path.join(project_dir, 'resource/pretrained_networks'),
)
self.lasot_dir = os.environ.get('LASOT_DIR', os.path.join(dataset_root, 'LaSOT/LaSOTBenchmark'))
self.got10k_dir = os.environ.get('GOT10K_DIR', os.path.join(dataset_root, 'got10k/train'))
self.lasot_lmdb_dir = os.environ.get('LASOT_LMDB_DIR', os.path.join(dataset_root, 'lasot_lmdb'))
self.got10k_lmdb_dir = os.environ.get('GOT10K_LMDB_DIR', os.path.join(dataset_root, 'got10k_lmdb'))
self.trackingnet_dir = os.environ.get('TRACKINGNET_DIR', os.path.join(dataset_root, 'trackingnet'))
self.trackingnet_lmdb_dir = os.environ.get('TRACKINGNET_LMDB_DIR', os.path.join(dataset_root, 'trackingnet_lmdb'))
self.coco_dir = os.environ.get('COCO_DIR', os.path.join(dataset_root, 'coco'))
self.coco_lmdb_dir = os.environ.get('COCO_LMDB_DIR', os.path.join(dataset_root, 'coco_lmdb'))
self.imagenet1k_dir = os.environ.get('IMAGENET1K_DIR', os.path.join(dataset_root, 'imagenet1k'))
self.imagenet22k_dir = os.environ.get('IMAGENET22K_DIR', os.path.join(dataset_root, 'imagenet22k'))
self.lvis_dir = os.environ.get('LVIS_DIR', '')
self.sbd_dir = os.environ.get('SBD_DIR', '')
self.imagenet_dir = os.environ.get('IMAGENET_DIR', os.path.join(dataset_root, 'vid'))
self.imagenet_lmdb_dir = os.environ.get('IMAGENET_LMDB_DIR', os.path.join(dataset_root, 'vid_lmdb'))
self.imagenetdet_dir = os.environ.get('IMAGENETDET_DIR', '')
self.ecssd_dir = os.environ.get('ECSSD_DIR', '')
self.hkuis_dir = os.environ.get('HKUIS_DIR', '')
self.msra10k_dir = os.environ.get('MSRA10K_DIR', '')
self.davis_dir = os.environ.get('DAVIS_DIR', '')
self.youtubevos_dir = os.environ.get('YOUTUBEVOS_DIR', '')
self.depthtrack_dir = os.environ.get('DEPTHTRACK_DIR', os.path.join(dataset_root, 'depthtrack/train'))
self.lasher_dir = os.environ.get('LASHER_DIR', os.path.join(dataset_root, 'lasher/trainingset'))
self.visevent_dir = os.environ.get('VISEVENT_DIR', os.path.join(dataset_root, 'visevent/train'))
self.refcoco_dir = os.environ.get('REFCOCO_DIR', os.path.join(dataset_root, 'refcoco'))
self.tnl2k_dir = os.environ.get('TNL2K_DIR', os.path.join(dataset_root, 'tnl2k/train'))
self.otb99_dir = os.environ.get('OTB99_DIR', os.path.join(dataset_root, 'otb_lang'))
|