File size: 2,854 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 | import os
from lib.test.evaluation.environment import EnvSettings
def local_env_settings():
settings = EnvSettings()
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')),
)
output_root = os.environ.get('SAVE_DIR', os.path.join(project_dir, 'output'))
settings.davis_dir = os.environ.get('DAVIS_DIR', '')
settings.got10k_lmdb_path = os.environ.get('GOT10K_LMDB_DIR', os.path.join(dataset_root, 'got10k_lmdb'))
settings.got10k_path = os.environ.get('GOT10K_DIR', os.path.join(dataset_root, 'got10k'))
settings.got_packed_results_path = os.environ.get('GOT_PACKED_RESULTS_PATH', '')
settings.got_reports_path = os.environ.get('GOT_REPORTS_PATH', '')
settings.lasot_extension_subset_path = os.environ.get('LASOT_EXTENSION_SUBSET_DIR', os.path.join(dataset_root, 'lasot_extension_subset'))
settings.lasot_lmdb_path = os.environ.get('LASOT_LMDB_DIR', os.path.join(dataset_root, 'lasot_lmdb'))
settings.lasot_path = os.environ.get('LASOT_TEST_DIR', os.path.join(dataset_root, 'lasot'))
settings.lasotlang_path = os.environ.get('LASOT_LANG_DIR', os.path.join(dataset_root, 'lasot'))
settings.network_path = os.environ.get('NETWORK_PATH', os.path.join(output_root, 'test/networks'))
settings.nfs_path = os.environ.get('NFS_DIR', os.path.join(dataset_root, 'nfs'))
settings.otb_path = os.environ.get('OTB_DIR', os.path.join(dataset_root, 'OTB2015'))
settings.otblang_path = os.environ.get('OTB_LANG_DIR', os.path.join(dataset_root, 'otb_lang'))
settings.prj_dir = project_dir
settings.result_plot_path = os.environ.get('RESULT_PLOT_PATH', os.path.join(output_root, 'test/result_plots'))
settings.results_path = os.environ.get('RESULTS_PATH', os.path.join(output_root, 'test/tracking_results'))
settings.save_dir = output_root
settings.segmentation_path = os.environ.get('SEGMENTATION_PATH', os.path.join(output_root, 'test/segmentation_results'))
settings.tc128_path = os.environ.get('TC128_DIR', os.path.join(dataset_root, 'TC128'))
settings.tn_packed_results_path = os.environ.get('TN_PACKED_RESULTS_PATH', '')
settings.tnl2k_path = os.environ.get('TNL2K_TEST_DIR', os.path.join(dataset_root, 'tnl2k/test'))
settings.tpl_path = os.environ.get('TPL_PATH', '')
settings.trackingnet_path = os.environ.get('TRACKINGNET_DIR', os.path.join(dataset_root, 'trackingnet'))
settings.uav_path = os.environ.get('UAV_DIR', os.path.join(dataset_root, 'UAV123'))
settings.vot_path = os.environ.get('VOT_DIR', os.path.join(dataset_root, 'VOT2019'))
settings.youtubevos_dir = os.environ.get('YOUTUBEVOS_DIR', '')
return settings
|