def initialize(self, opt): # SYNTHIA as dataset 1 # GTAV as dataset 2 self.opt = opt self.root = opt.dataroot self.dir_A_1 = os.path.join(opt.dataroot, 'synthia', 'RGB') self.dir_A_2 = os.path.join(opt.dataroot, 'gta5', 'images') self.dir_B = os.path.join(opt.dataroot, 'cityscapes', 'leftImg8bit') self.dir_A_label_1 = os.path.join(opt.dataroot, 'synthia', 'GT', 'parsed_LABELS') self.dir_A_label_2 = os.path.join(opt.dataroot, 'gta5', 'labels') self.A_paths_1 = make_dataset(self.dir_A_1) self.A_paths_2 = make_dataset(self.dir_A_2) self.B_paths = make_dataset(self.dir_B) self.A_paths_1 = sorted(self.A_paths_1) self.A_paths_2 = sorted(self.A_paths_2) self.B_paths = sorted(self.B_paths) self.A_size_1 = len(self.A_paths_1) self.A_size_2 = len(self.A_paths_2) self.B_size = len(self.B_paths) self.A_labels_1 = make_dataset(self.dir_A_label_1) self.A_labels_2 = make_dataset(self.dir_A_label_2) self.A_labels_1 = sorted(self.A_labels_1) self.A_labels_2 = sorted(self.A_labels_2) self.transform = get_transform(opt) self.label_transform = get_label_transform(opt)
def initialize(self, opt): self.opt = opt self.root = opt.dataroot self.dir_A = os.path.join(opt.dataroot, 'synthia', 'RGB') self.dir_B = os.path.join(opt.dataroot, 'cityscapes', 'leftImg8bit') self.dir_A_label = os.path.join(opt.dataroot, 'synthia', 'GT', 'parsed_LABELS') self.dir_B_label = os.path.join(opt.dataroot, 'cityscapes', 'gtFine') self.A_paths = make_dataset(self.dir_A) self.B_paths = make_dataset(self.dir_B) self.A_paths = sorted(self.A_paths) self.B_paths = sorted(self.B_paths) self.A_size = len(self.A_paths) self.B_size = len(self.B_paths) self.A_labels = make_dataset(self.dir_A_label) self.B_labels = make_cs_labels(self.dir_B_label) self.A_labels = sorted(self.A_labels) self.B_labels = sorted(self.B_labels) self.transform = get_transform(opt) self.label_transform = get_label_transform(opt)