Map-Detect / data_config.py
justchenhao's picture
BIT
d43892c
Raw
History Blame Contribute Delete
612 Bytes
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)