def CreateTrgDataLoader(args):
if args.set == 'train' or args.set == 'trainval':
target_dataset = cityscapesDataSetLabel(
args.data_dir_target,
args.data_list_target,
crop_size=image_sizes['cityscapes'],
mean=IMG_MEAN,
max_iters=args.num_steps * args.batch_size,
set=args.set)
else:
target_dataset = cityscapesDataSet(
args.data_dir_target,
args.data_list_target,
crop_size=cs_size_test['cityscapes'],
mean=IMG_MEAN,
set=args.set)
if args.set == 'train' or args.set == 'trainval':
target_dataloader = data.DataLoader(target_dataset,
batch_size=args.batch_size,
shuffle=True,
num_workers=args.num_workers,
pin_memory=True)
else:
target_dataloader = data.DataLoader(target_dataset,
batch_size=1,
shuffle=False,
pin_memory=True)
return target_dataloader
def CreateTrgDataLoader(args):
if args.source == 'triangle':
# for simple triangle dataset
target_dataset = triangleDatasetTgt(args.data_dir_target,
args.data_list_target,
max_iters=args.num_steps *
args.batch_size,
crop_size=image_sizes['triangle'],
mean=IMG_MEAN)
target_dataloader = data.DataLoader(target_dataset,
batch_size=args.batch_size,
shuffle=True,
num_workers=args.num_workers,
pin_memory=True)
return target_dataloader
if args.data_label_folder_target is not None:
target_dataset = cityscapesDataSetLabel(
args.data_dir_target,
args.data_list_target,
max_iters=args.num_steps * args.batch_size,
crop_size=image_sizes['cityscapes'],
mean=IMG_MEAN,
set=args.set,
label_folder=args.data_label_folder_target)
else:
if args.set == 'train':
target_dataset = cityscapesDataSet(
args.data_dir_target,
args.data_list_target,
max_iters=args.num_steps * args.batch_size,
crop_size=image_sizes['cityscapes'],
mean=IMG_MEAN,
set=args.set)
else:
target_dataset = cityscapesDataSet(
args.data_dir_target,
args.data_list_target,
crop_size=image_sizes['cityscapes'],
mean=IMG_MEAN,
set=args.set)
if args.set == 'train':
target_dataloader = data.DataLoader(target_dataset,
batch_size=args.batch_size,
shuffle=True,
num_workers=args.num_workers,
pin_memory=True)
else:
target_dataloader = data.DataLoader(target_dataset,
batch_size=1,
shuffle=False,
pin_memory=True)
return target_dataloader
def CreateTrgDataLoader(args, mode='train'):
if mode == 'train':
target_dataset = cityscapesDataSetLabel(
args.data_dir_target,
args.data_list_target_train,
crop_size=image_sizes['cityscapes'],
mean=IMG_MEAN,
set=mode)
elif mode == 'val':
target_dataset = cityscapesDataSetLabel(
args.data_dir_target,
args.data_list_target_val,
crop_size=cs_size_test['cityscapes'],
mean=IMG_MEAN,
set=mode)
else:
raise Exception(
"Argument set has not entered properly. Options are train or eval."
)
if mode == 'train':
target_dataloader = data.DataLoader(target_dataset,
batch_size=args.batch_size,
shuffle=True,
num_workers=args.num_workers,
pin_memory=True)
elif mode == 'val':
target_dataloader = data.DataLoader(target_dataset,
batch_size=args.batch_size,
shuffle=False,
num_workers=args.num_workers,
pin_memory=True)
else:
raise Exception(
"Argument set has not entered properly. Options are train or eval."
)
return target_dataloader