Add ImageNet dataset (placeholder)

sungnyun · Jan 20, 2022 · d3d858d · d3d858d
1 parent d32c3e9
commit d3d858d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/configs.py b/configs.py
@@ -4,6 +4,7 @@
 miniImageNet_test_path          = '/data/cdfsl/miniImagenet_test'
 tieredImageNet_path           	= '/data/cdfsl/tieredImagenet/train'
 tieredImageNet_test_path        = '/data/cdfsl/tieredImagenet/test'
+ImageNet_path                   = '/data/cdfsl/Imagenet/train'
 DTD_path 						= '/ssd/dtd/images/'
 
 ISIC_path = "/data/cdfsl/ISIC"

diff --git a/datasets/datasets.py b/datasets/datasets.py
@@ -39,6 +39,13 @@ def __init__(self, root=tieredImageNet_test_path, *args, **kwargs):
         super().__init__(root=root, *args, **kwargs)
 
 
+class ImageNetDataset(ImageFolder):
+    name = "ImageNet"
+
+    def __init__(self, root=ImageNet_path, *args, **kwargs):
+        super().__init__(root=root, *args, **kwargs)
+
+
 class CropDiseaseDataset(ImageFolder):
     name = "CropDisease"