init

datasets/__init__.py

@@ -0,0 +1,33 @@
+from .oxford_pets import OxfordPets
+from .eurosat import EuroSAT
+from .ucf101 import UCF101
+from .sun397 import SUN397
+from .caltech101 import Caltech101
+from .dtd import DescribableTextures
+from .fgvc import FGVCAircraft
+from .food101 import Food101
+from .oxford_flowers import OxfordFlowers
+from .stanford_cars import StanfordCars
+from .imagenet import ImageNet
+from .caltech101_tsne import Caltech101_TSNE
+
+
+
+dataset_list = {
+                "oxford_pets": OxfordPets,
+                "eurosat": EuroSAT,
+                "ucf101": UCF101,
+                "sun397": SUN397,
+                "caltech101": Caltech101,
+                "dtd": DescribableTextures,
+                "fgvc": FGVCAircraft,
+                "food101": Food101,
+                "oxford_flowers": OxfordFlowers,
+                "stanford_cars": StanfordCars,
+                "caltech101_tsne": Caltech101_TSNE,
+                "imagenet":ImageNet,
+                }
+
+
+def build_dataset(dataset, root_path, shots):
+    return dataset_list[dataset](root_path, shots)