-
Notifications
You must be signed in to change notification settings - Fork 2
/
data_config.py
30 lines (26 loc) · 936 Bytes
/
data_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
class DataConfig:
data_name = ""
root_dir = ""
label_transform = "norm"
def get_data_config(self, data_name):
self.data_name = data_name
if data_name == 'LEVIR+':
self.root_dir = 'D:/data/CD/LEVIR+/' # Your Path
elif data_name == 'LEVIR':
self.root_dir = 'D:/data/CD/LEVIR/'
elif data_name == 'WHU':
self.root_dir = 'D:/data/CD/WHU/'
elif data_name == 'GZ':
self.root_dir = 'D:/data/CD/GZ/'
elif data_name == 'SYSU':
self.root_dir = 'D:/data/CD/SYSU/'
elif data_name == 'quick_start':
self.root_dir = './samples/'
else:
raise TypeError('%s has not defined' % data_name)
return self
if __name__ == '__main__':
data = DataConfig().get_data_config(data_name='LEVIR')
# print(data.data_name)
# print(data.root_dir)
# print(data.label_transform)