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/train/admin/local.py
+++ b/lib/train/admin/local.py
@@ -0,0 +1,24 @@
+class EnvironmentSettings:
+    def __init__(self):
+        self.workspace_dir = '/home/baiyifan/code/2stage_update_intrain'    # Base directory for saving network checkpoints.
+        self.tensorboard_dir = '/home/baiyifan/code/2stage/tensorboard'    # Directory for tensorboard files.
+        self.pretrained_networks = '/home/baiyifan/code/2stage/pretrained_networks'
+        self.lasot_dir = '/home/baiyifan/LaSOT/LaSOTBenchmark'
+        self.got10k_dir = '/home/baiyifan/GOT-10k/train'
+        self.got10k_val_dir = '/home/baiyifan/GOT-10k/val'
+        self.lasot_lmdb_dir = '/home/baiyifan/code/2stage/data/lasot_lmdb'
+        self.got10k_lmdb_dir = '/home/baiyifan/code/2stage/data/got10k_lmdb'
+        self.trackingnet_dir = '/ssddata/TrackingNet/all_zip'
+        self.trackingnet_lmdb_dir = '/home/baiyifan/code/2stage/data/trackingnet_lmdb'
+        self.coco_dir = '/home/baiyifan/coco'
+        self.coco_lmdb_dir = '/home/baiyifan/code/2stage/data/coco_lmdb'
+        self.lvis_dir = ''
+        self.sbd_dir = ''
+        self.imagenet_dir = '/home/baiyifan/code/2stage/data/vid'
+        self.imagenet_lmdb_dir = '/home/baiyifan/code/2stage/data/vid_lmdb'
+        self.imagenetdet_dir = ''
+        self.ecssd_dir = ''
+        self.hkuis_dir = ''
+        self.msra10k_dir = ''
+        self.davis_dir = ''
+        self.youtubevos_dir = ''