data_config.py

class DataConfig:
    data_name = ""
    root_dir = ""
    label_transform = ""
    def get_data_config(self, data_name):
        self.data_name = data_name
        if data_name == 'LEVIR':
            self.label_transform = "norm"
            self.root_dir = '../../data/LEVIR-CD256'
        elif data_name == 'DSIFN':
            self.label_transform = "norm"
            self.root_dir =  '../dataset/CD/DSIFN-CD-256/'
        elif data_name == 'WHU':
            self.label_transform = "norm"
            self.root_dir = '../dataset/CD/WHU-CD-256/'
        elif data_name == 'SYSU':
            self.label_transform = "norm"
            self.root_dir = '../dataset/CD/SYSU-CD-256/'
        elif data_name == 'CDD':
            self.label_transform = "norm"
            self.root_dir =  '../../data/CDD2'
        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)