forked from fupiao1998/TransformerSOD
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathpath_config.py
64 lines (63 loc) · 4.5 KB
/
path_config.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
def get_path_dict(hostname, task):
path_dict = {}
if hostname == 'LabPC2':
if task == 'COD':
path_dict['image_root'] = '/home1/datasets/SOD_COD/COD/camouflage/COD_train/Imgs/'
path_dict['gt_root'] = '/home1/datasets/SOD_COD/COD/camouflage/COD_train/GT/'
path_dict['test_dataset_root'] = '/home1/datasets/SOD_COD/COD/COD_test/'
elif task == 'SOD':
path_dict['image_root'] = '/home1/maoyuxin/datasets/SOD/DUTS/img/'
path_dict['gt_root'] = '/home1/maoyuxin/datasets/SOD/DUTS/gt/'
path_dict['test_dataset_root'] = '/home1/maoyuxin/datasets/SOD/SOD_RGB/'
elif task == 'RGBD-SOD':
path_dict['image_root'] = '/home1/maoyuxin/datasets/SOD/RGBD_SOD/train/RGB/'
path_dict['gt_root'] = '/home1/maoyuxin/datasets/SOD/RGBD_SOD/train/GT/'
path_dict['depth_root'] = '/home1/maoyuxin/datasets/SOD/RGBD_SOD/train/depth/'
path_dict['test_dataset_root'] = '/home1/maoyuxin/datasets/SOD/RGBD_SOD/test/'
elif task == 'Weak-RGB-SOD':
path_dict['image_root'] = '/home1/datasets/SOD_COD/Scribble_SOD/img/'
path_dict['gt_root'] = '/home1/datasets/SOD_COD/Scribble_SOD/gt/'
path_dict['mask_root'] = '/home1/datasets/SOD_COD/Scribble_SOD/mask/'
path_dict['gray_root'] = '/home1/datasets/SOD_COD/Scribble_SOD/gray/'
path_dict['test_dataset_root'] = '/home1/datasets/SOD_COD/SOD_RGB/'
elif hostname == 'gpu6':
if task == 'COD':
path_dict['image_root'] = '/home1/datasets/SOD_COD/COD/camouflage/COD_train/Imgs/'
path_dict['gt_root'] = '/home1/datasets/SOD_COD/COD/camouflage/COD_train/GT/'
path_dict['test_dataset_root'] = '/home1/datasets/SOD_COD/COD/COD_test/'
elif task == 'SOD':
path_dict['image_root'] = '/home/maoyuxin/dataset/SOD_COD/DUTS/img/'
path_dict['gt_root'] = '/home/maoyuxin/dataset/SOD_COD/DUTS/gt/'
path_dict['test_dataset_root'] = '/home2/dataset/maoyuxin/SOD_COD/SOD_RGB/'
elif task == 'RGBD-SOD':
path_dict['image_root'] = '/home/maoyuxin/dataset/SOD_COD/RGBD_SOD/train/RGB/'
path_dict['gt_root'] = '/home/maoyuxin/dataset/SOD_COD/RGBD_SOD/train/GT/'
path_dict['depth_root'] = '/home/maoyuxin/dataset/SOD_COD/RGBD_SOD/train/depth/'
path_dict['test_dataset_root'] = '/home/maoyuxin/dataset/SOD_COD/RGBD_SOD/test/'
elif task == 'Weak-RGB-SOD':
path_dict['image_root'] = '/home1/datasets/SOD_COD/Scribble_SOD/img/'
path_dict['gt_root'] = '/home1/datasets/SOD_COD/Scribble_SOD/gt/'
path_dict['mask_root'] = '/home1/datasets/SOD_COD/Scribble_SOD/mask/'
path_dict['gray_root'] = '/home1/datasets/SOD_COD/Scribble_SOD/gray/'
path_dict['test_dataset_root'] = '/home2/dataset/maoyuxin/SOD_COD/SOD_RGB/'
elif hostname == 'server10':
if task == 'COD':
path_dict['image_root'] = '/data/local_userdata/maoyuxin/SOD/SOD_COD/COD/camouflage/COD_train/Imgs/'
path_dict['gt_root'] = '/data/local_userdata/maoyuxin/SOD/SOD_COD/COD/camouflage/COD_train/GT/'
path_dict['test_dataset_root'] = '/data/local_userdata/maoyuxin/SOD/SOD_COD/COD/COD_test/'
elif task == 'SOD':
path_dict['image_root'] = '/data/local_userdata/maoyuxin/SOD/SOD_COD/DUTS/img/'
path_dict['gt_root'] = '/data/local_userdata/maoyuxin/SOD/SOD_COD/DUTS/gt/'
path_dict['test_dataset_root'] = '/data/local_userdata/maoyuxin/SOD/SOD_COD/SOD_RGB/'
elif task == 'RGBD-SOD':
path_dict['image_root'] = '/data/local_userdata/maoyuxin/SOD/SOD_COD/RGBD_SOD/train/RGB/'
path_dict['gt_root'] = '/data/local_userdata/maoyuxin/SOD/SOD_COD/RGBD_SOD/train/GT/'
path_dict['depth_root'] = '/data/local_userdata/maoyuxin/SOD/SOD_COD/RGBD_SOD/train/depth/'
path_dict['test_dataset_root'] = '/data/local_userdata/maoyuxin/SOD/SOD_COD/RGBD_SOD/test/'
elif task == 'Weak-RGB-SOD':
path_dict['image_root'] = '/data/local_userdata/maoyuxin/SOD/SOD_COD/Scribble_SOD/img/'
path_dict['gt_root'] = '/data/local_userdata/maoyuxin/SOD/SOD_COD/Scribble_SOD/gt/'
path_dict['mask_root'] = '/data/local_userdata/maoyuxin/SOD/SOD_COD/Scribble_SOD/mask/'
path_dict['gray_root'] = '/data/local_userdata/maoyuxin/SOD/SOD_COD/Scribble_SOD/gray/'
path_dict['test_dataset_root'] = '/data/local_userdata/maoyuxin/SOD/SOD_COD/SOD_RGB/'
return path_dict