def get_data(name, data_dir, height, width, batch_size, workers, num_instances, iters=200):
root = osp.join(data_dir, name)
dataset = datasets.create(name, root)
normalizer = T.Normalize(mean=[0.485, 0.456, 0.406],
std=[0.229, 0.224, 0.225])
train_set = dataset.train
num_classes = dataset.num_train_pids
train_transformer = T.Compose([
T.Resize((height, width), interpolation=3),
T.RandomHorizontalFlip(p=0.5),
T.Pad(10),
T.RandomCrop((height, width)),
T.ToTensor(),
normalizer
])
test_transformer = T.Compose([
T.Resize((height, width), interpolation=3),
T.ToTensor(),
normalizer
])
rmgs_flag = num_instances > 0
if rmgs_flag:
sampler = RandomMultipleGallerySampler(train_set, num_instances)
else:
sampler = None
train_loader = IterLoader(
DataLoader(Preprocessor(train_set, root=dataset.images_dir,
transform=train_transformer),
batch_size=batch_size, num_workers=workers, sampler=sampler,
shuffle=not rmgs_flag, pin_memory=True, drop_last=True), length=iters)
test_loader = DataLoader(
Preprocessor(list(set(dataset.query) | set(dataset.gallery)),
root=dataset.images_dir, transform=test_transformer),
batch_size=batch_size, num_workers=workers,
shuffle=False, pin_memory=True)
return dataset, num_classes, train_loader, test_loader
def get_test_data(name, data_dir, height, width, batch_size, workers):
dataset = datasets.create(name, data_dir)
normalizer = T.Normalize(mean=[0.485, 0.456, 0.406],
std=[0.229, 0.224, 0.225])
test_transformer = T.Compose(
[T.Resize((height, width), interpolation=3),
T.ToTensor(), normalizer])
test_loader = DataLoader(Preprocessor(
list(set(dataset.query) | set(dataset.gallery)),
root=None,
transform=test_transformer),
batch_size=batch_size,
num_workers=workers,
shuffle=False,
pin_memory=True)
return dataset, test_loader
def get_data(name, data_dir, height, width, batch_size, workers):
normalizer = T.Normalize(mean=[0.485, 0.456, 0.406],
std=[0.229, 0.224, 0.225])
test_transforms = T.Compose([
T.Resize((height, width), interpolation=3),
T.ToTensor(),
normalizer
])
dataset = datasets.create(name, data_dir)
query_set = ImageDataset(dataset.query, test_transforms)
gallery_set = ImageDataset(dataset.gallery, test_transforms)
query_loader = DataLoader(
query_set, batch_size=batch_size, shuffle=False,
collate_fn=val_collate_fn, num_workers=workers, pin_memory=True
)
gallery_loader = DataLoader(
gallery_set, batch_size=batch_size, shuffle=False,
collate_fn=val_collate_fn, num_workers=workers, pin_memory=True
)
return query_loader, gallery_loader
def get_data(name, data_dir):
root = osp.join(data_dir, name)
dataset = datasets.create(name, root)
return dataset
def get_data(name, data_dir):
dataset = datasets.create(name, data_dir)
return dataset
