diff --git a/dataset/cifar100.py b/dataset/cifar100.py index 7f59aafe..749ad470 100644 --- a/dataset/cifar100.py +++ b/dataset/cifar100.py @@ -35,8 +35,29 @@ def get_data_folder(): return data_folder +class CIFAR100BackCompat(datasets.CIFAR100): + """ + CIFAR100Instance+Sample Dataset + """ + + @property + def train_labels(self): + return self.targets + + @property + def test_labels(self): + return self.targets + + @property + def train_data(self): + return self.data + + @property + def test_data(self): + return self.data + -class CIFAR100Instance(datasets.CIFAR100): +class CIFAR100Instance(CIFAR100BackCompat): """CIFAR100Instance Dataset. """ def __getitem__(self, index): @@ -106,7 +127,7 @@ def get_cifar100_dataloaders(batch_size=128, num_workers=8, is_instance=False): return train_loader, test_loader -class CIFAR100InstanceSample(datasets.CIFAR100): +class CIFAR100InstanceSample(CIFAR100BackCompat): """ CIFAR100Instance+Sample Dataset """