Files
Grounded-SAM-2/lib/train/admin/local.py
2024-11-19 22:12:54 -08:00

25 lines
1.3 KiB
Python

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