| import importlib |
| import os |
|
|
|
|
| class EnvSettings: |
| def __init__(self): |
| test_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) |
|
|
| self.results_path = '{}/tracking_results/'.format(test_path) |
| self.segmentation_path = '{}/segmentation_results/'.format(test_path) |
| self.network_path = '{}/networks/'.format(test_path) |
| self.result_plot_path = '{}/result_plots/'.format(test_path) |
| self.otb_path = '' |
| self.nfs_path = '' |
| self.uav_path = '' |
| self.tpl_path = '' |
| self.vot_path = '' |
| self.got10k_path = '' |
| self.lasot_path = '' |
| self.trackingnet_path = '' |
| self.davis_dir = '' |
| self.youtubevos_dir = '' |
|
|
| self.got_packed_results_path = '' |
| self.got_reports_path = '' |
| self.tn_packed_results_path = '' |
|
|
|
|
| def create_default_local_file(): |
| comment = {'results_path': 'Where to store tracking results', |
| 'network_path': 'Where tracking networks are stored.'} |
|
|
| path = os.path.join(os.path.dirname(__file__), 'local.py') |
| with open(path, 'w') as f: |
| settings = EnvSettings() |
|
|
| f.write('from test.evaluation.environment import EnvSettings\n\n') |
| f.write('def local_env_settings():\n') |
| f.write(' settings = EnvSettings()\n\n') |
| f.write(' # Set your local paths here.\n\n') |
|
|
| for attr in dir(settings): |
| comment_str = None |
| if attr in comment: |
| comment_str = comment[attr] |
| attr_val = getattr(settings, attr) |
| if not attr.startswith('__') and not callable(attr_val): |
| if comment_str is None: |
| f.write(' settings.{} = \'{}\'\n'.format(attr, attr_val)) |
| else: |
| f.write(' settings.{} = \'{}\' # {}\n'.format(attr, attr_val, comment_str)) |
| f.write('\n return settings\n\n') |
|
|
|
|
| class EnvSettings_ITP: |
| def __init__(self, workspace_dir, data_dir, save_dir): |
| self.prj_dir = workspace_dir |
| self.save_dir = save_dir |
| self.results_path = os.path.join(save_dir, 'test/tracking_results') |
| self.segmentation_path = os.path.join(save_dir, 'test/segmentation_results') |
| self.network_path = os.path.join(save_dir, 'test/networks') |
| self.result_plot_path = os.path.join(save_dir, 'test/result_plots') |
| self.otb_path = os.path.join(data_dir, 'OTB2015') |
| self.nfs_path = os.path.join(data_dir, 'nfs') |
| self.uav_path = os.path.join(data_dir, 'UAV123') |
| self.tc128_path = os.path.join(data_dir, 'TC128') |
| self.tpl_path = '' |
| self.vot_path = os.path.join(data_dir, 'VOT2019') |
| self.got10k_path = os.path.join(data_dir, 'got10k') |
| self.got10k_lmdb_path = os.path.join(data_dir, 'got10k_lmdb') |
| self.lasot_path = os.path.join(data_dir, 'lasot') |
| self.lasot_lmdb_path = os.path.join(data_dir, 'lasot_lmdb') |
| self.lasot_extension_subset_path = os.path.join(data_dir, 'lasot_extension_subset') |
| self.trackingnet_path = os.path.join(data_dir, 'trackingnet') |
| self.tnl2k_path = os.path.join(data_dir, 'tnl2k/test') |
| self.davis_dir = '' |
| self.youtubevos_dir = '' |
| self.otblang_path = os.path.join(data_dir, 'otb_lang') |
| self.lasotlang_path = os.path.join(data_dir, 'lasot') |
|
|
| self.got_packed_results_path = '' |
| self.got_reports_path = '' |
| self.tn_packed_results_path = '' |
|
|
|
|
| def create_default_local_file_ITP_test(workspace_dir, data_dir, save_dir): |
| comment = {'results_path': 'Where to store tracking results', |
| 'network_path': 'Where tracking networks are stored.'} |
|
|
| path = os.path.join(os.path.dirname(__file__), 'local.py') |
| with open(path, 'w') as f: |
| settings = EnvSettings_ITP(workspace_dir, data_dir, save_dir) |
|
|
| f.write('from lib.test.evaluation.environment import EnvSettings\n\n') |
| f.write('def local_env_settings():\n') |
| f.write(' settings = EnvSettings()\n\n') |
| f.write(' # Set your local paths here.\n\n') |
|
|
| for attr in dir(settings): |
| comment_str = None |
| if attr in comment: |
| comment_str = comment[attr] |
| attr_val = getattr(settings, attr) |
| if not attr.startswith('__') and not callable(attr_val): |
| if comment_str is None: |
| f.write(' settings.{} = \'{}\'\n'.format(attr, attr_val)) |
| else: |
| f.write(' settings.{} = \'{}\' # {}\n'.format(attr, attr_val, comment_str)) |
| f.write('\n return settings\n\n') |
|
|
|
|
| def env_settings(): |
| env_module_name = 'lib.test.evaluation.local' |
| try: |
| env_module = importlib.import_module(env_module_name) |
| return env_module.local_env_settings() |
| except: |
| env_file = os.path.join(os.path.dirname(__file__), 'local.py') |
|
|
| |
| create_default_local_file() |
| raise RuntimeError('YOU HAVE NOT SETUP YOUR local.py!!!\n Go to "{}" and set all the paths you need. ' |
| 'Then try to run again.'.format(env_file)) |
