Merge branch 'tinyimagenet_dataset' into 'main'

moreib · moreib · commit c0daf6321196 · 2025-02-11T09:18:37.000Z
TinyImageNet Dataset

See merge request es/ai/hannah/hannah!422
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"
 ]
diff --git a/hannah/datasets/vision/imagenet.py b/hannah/datasets/vision/imagenet.py
