| 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')) | |