def loading_data():
_, _, img_transform, gt_transform, restore_transform = data_transforms(
cfg_data=cfg_data)
# train loader
train_set = Mall(cfg_data.DATA_PATH + '/train',
'train',
main_transform=None,
img_transform=img_transform,
gt_transform=gt_transform)
train_loader = DataLoader(train_set,
batch_size=cfg_data.TRAIN_BATCH_SIZE,
num_workers=0,
shuffle=True,
drop_last=True)
# test loader
test_set = Mall(cfg_data.DATA_PATH + '/test',
'test',
main_transform=None,
img_transform=img_transform,
gt_transform=gt_transform)
test_loader = DataLoader(test_set,
batch_size=cfg_data.VAL_BATCH_SIZE,
num_workers=0,
shuffle=False,
drop_last=False)
return train_loader, test_loader, restore_transform
def loading_data():
train_main_transform, val_main_transform, img_transform, gt_transform, restore_transform = data_transforms(
cfg_data=cfg_data)
train_set = City(cfg_data.DATA_PATH + '/train',
main_transform=train_main_transform,
img_transform=img_transform,
gt_transform=gt_transform)
train_loader = DataLoader(train_set,
batch_size=cfg_data.TRAIN_BATCH_SIZE,
num_workers=10,
shuffle=True,
drop_last=False)
val_set = City(cfg_data.DATA_PATH + '/test',
main_transform=val_main_transform,
img_transform=img_transform,
gt_transform=gt_transform)
val_loader = DataLoader(val_set,
batch_size=cfg_data.VAL_BATCH_SIZE,
num_workers=10,
shuffle=False,
drop_last=False)
return train_loader, val_loader, restore_transform
def loading_data(scene_folder=None):
train_main_transform, val_main_transform, img_transform, gt_transform, restore_transform = data_transforms(
cfg_data=cfg_data)
# train loader
train_set = FDST(os.path.join(cfg_data.DATA_PATH, 'train'),
main_transform=train_main_transform,
img_transform=img_transform,
gt_transform=gt_transform,
scene_folder=scene_folder)
train_loader = DataLoader(train_set,
batch_size=cfg_data.TRAIN_BATCH_SIZE,
num_workers=10,
shuffle=True,
drop_last=True)
# validation loader
val_path = os.path.join(cfg_data.DATA_PATH, 'test')
val_set = FDST(val_path,
main_transform=val_main_transform,
img_transform=img_transform,
gt_transform=gt_transform,
scene_folder=scene_folder)
val_loader = DataLoader(val_set,
batch_size=cfg_data.VAL_BATCH_SIZE,
num_workers=10,
shuffle=False,
drop_last=True)
return train_loader, val_loader, restore_transform
def loading_data(k_shot=1, scene_folder=None):
train_main_transform, val_main_transform, img_transform, gt_transform, restore_transform = data_transforms(
cfg_data=cfg_data)
VAL_FOLDERS = [os.path.join(sf, scene_folder)
for sf in cfg_data.VAL_FOLDERS]
TRAIN_FOLDERS = [os.path.join(sf, scene_folder)
for sf in cfg_data.TRAIN_FOLDERS]
# train loader
train_set = PETS(
os.path.join(
cfg_data.DATA_PATH,
'train'),
main_transform=train_main_transform,
img_transform=img_transform,
gt_transform=gt_transform,
k_shot=k_shot,
val_folders=TRAIN_FOLDERS)
train_loader = DataLoader(
train_set,
batch_size=cfg_data.TRAIN_BATCH_SIZE,
num_workers=1,
shuffle=True,
drop_last=True)
# test
test_path = os.path.join(cfg_data.DATA_PATH, 'test')
test_set = PETS(
test_path,
main_transform=val_main_transform,
img_transform=img_transform,
gt_transform=gt_transform,
val_folders=VAL_FOLDERS)
test_loader = DataLoader(
test_set,
batch_size=cfg_data.VAL_BATCH_SIZE,
num_workers=1,
shuffle=False,
drop_last=True)
return train_loader, test_loader, restore_transform
