init commit of samurai

2024-11-19 22:12:54 -08:00
parent f65f4ba181
commit c17e4cecc0
679 changed files with 123982 additions and 0 deletions
--- a/lib/test/evaluation/environment.py
+++ b/lib/test/evaluation/environment.py
@@ -0,0 +1,124 @@
+import importlib
+import os
+
+
+class EnvSettings:
+    def __init__(self):
+        test_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '..'))
+
+        self.results_path = '{}/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, 'otb')
+        self.nfs_path = os.path.join(data_dir, 'nfs')
+        self.uav_path = os.path.join(data_dir, 'uav')
+        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.trackingnet_path = os.path.join(data_dir, 'trackingnet')
+        self.vot18_path = os.path.join(data_dir, 'vot2018')
+        self.vot22_path = os.path.join(data_dir, 'vot2022')
+        self.itb_path = os.path.join(data_dir, 'itb')
+        self.tnl2k_path = os.path.join(data_dir, 'tnl2k')
+        self.lasot_extension_subset_path_path = os.path.join(data_dir, 'lasot_extension_subset')
+        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_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 a default file
+        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))