Merge branch 'tinyimagenet_dataset' into 'main'

TinyImageNet Dataset See merge request es/ai/hannah/hannah!422
ekut-es · Feb 11, 2025 · c0daf63 · c0daf63
2 parents cd6c200 + 3f37b88
commit c0daf63
Show file tree

Hide file tree

Showing 3 changed files with 398 additions and 0 deletions.
diff --git a/hannah/conf/dataset/tinyimagenet.yaml b/hannah/conf/dataset/tinyimagenet.yaml
@@ -0,0 +1,4 @@
+data_folder: ${hydra:runtime.cwd}/datasets/
+cls: hannah.datasets.vision.TinyImageNet
+dataset: tiny-imagenet-200
+val_percent: 0.1
diff --git a/hannah/datasets/vision/__init__.py b/hannah/datasets/vision/__init__.py
@@ -24,6 +24,7 @@
 from .ri_capsule import RICapsuleDataset
 from .mnist import MNISTDataset
 from .svhn import SVHNDataset
+from .imagenet import TinyImageNet
 
 __all__ = [
     "DresdenCapsuleDataset",
@@ -35,4 +36,5 @@
     "RICapsuleDataset",
     "MNISTDataset",
     "SVHNDataset",
+    "TinyImageNet"
 ]