def test_transforms_groups_base_usage(self):
original_dataset = MNIST('./data/mnist', download=True)
dataset = AvalancheDataset(
original_dataset,
transform_groups=dict(train=(ToTensor(), None),
eval=(None, Lambda(lambda t: float(t)))))
x, y, _ = dataset[0]
self.assertIsInstance(x, Tensor)
self.assertIsInstance(y, int)
dataset_test = dataset.eval()
x2, y2, _ = dataset_test[0]
x3, y3, _ = dataset[0]
self.assertIsInstance(x2, Image)
self.assertIsInstance(y2, float)
self.assertIsInstance(x3, Tensor)
self.assertIsInstance(y3, int)
dataset_train = dataset.train()
dataset.transform = None
x4, y4, _ = dataset_train[0]
x5, y5, _ = dataset[0]
self.assertIsInstance(x4, Tensor)
self.assertIsInstance(y4, int)
self.assertIsInstance(x5, Image)
self.assertIsInstance(y5, int)
def test_freeze_transforms(self):
original_dataset = MNIST('./data/mnist', download=True)
x, y = original_dataset[0]
dataset = AvalancheDataset(original_dataset, transform=ToTensor())
dataset_frozen = dataset.freeze_transforms()
dataset_frozen.transform = None
x2, y2, _ = dataset_frozen[0]
self.assertIsInstance(x2, Tensor)
self.assertIsInstance(y2, int)
self.assertTrue(torch.equal(ToTensor()(x), x2))
self.assertEqual(y, y2)
dataset.transform = None
x2, y2, _ = dataset[0]
self.assertIsInstance(x2, Image)
x2, y2, _ = dataset_frozen[0]
self.assertIsInstance(x2, Tensor)
def test_freeze_transforms_chain(self):
original_dataset = MNIST('./data/mnist',
download=True,
transform=ToTensor())
x, *_ = original_dataset[0]
self.assertIsInstance(x, Tensor)
dataset_transform = AvalancheDataset(original_dataset,
transform=ToPILImage())
x, *_ = dataset_transform[0]
self.assertIsInstance(x, Image)
dataset_frozen = dataset_transform.freeze_transforms()
x2, *_ = dataset_frozen[0]
self.assertIsInstance(x2, Image)
dataset_transform.transform = None
x2, *_ = dataset_transform[0]
self.assertIsInstance(x2, Tensor)
dataset_frozen.transform = ToTensor()
x2, *_ = dataset_frozen[0]
self.assertIsInstance(x2, Tensor)
dataset_frozen2 = dataset_frozen.freeze_transforms()
x2, *_ = dataset_frozen2[0]
self.assertIsInstance(x2, Tensor)
dataset_frozen.transform = None
x2, *_ = dataset_frozen2[0]
self.assertIsInstance(x2, Tensor)
x2, *_ = dataset_frozen[0]
self.assertIsInstance(x2, Image)