Spaces:
Running on Zero
Running on Zero
File size: 612 Bytes
d43892c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
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 = 'path to the root of LEVIR-CD dataset'
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)
|