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