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