def get_dataset(name):
     if name == "MNIST":
         return data.MNIST()
     elif name == "CIFAR100":
         return data.CIFAR100()
     elif name == "CIFAR10":
         return data.CIFAR10()
     else:
         print("Unsupported Dataset")
         assert False
Example #2
0
 def get_dataset(name):
     if name == 'CIFAR10':
         return data.CIFAR10()
     elif name == "CIFAR100":
         return data.CIFAR100()
     elif name == "Imagenet":
         return data.Imagenet()
     elif name == "VggFace2_1K":
         return data.VggFace2_1K()
     elif name == "VggFace2_5K":
         return data.VggFace2_5K()
     elif name == "Google_Landmark_v2_1K":
         return data.Google_Landmark_v2_1K()
     elif name == "Google_Landmark_v2_10K":
         return data.Google_Landmark_v2_10K()
Example #3
0
 def get_dataset(name):
     if name == 'CIFAR100':
         return data.CIFAR100()
     elif name == "TinyImagenet":
         return data.TinyImagenet()
 def get_dataset(name):
     if name == "CIFAR100":
         return data.CIFAR100()
     elif name == 'MNIST':
         return data.MNIST()