Merge pull request #60 from FreeStyleFreeLunch/main

add fer2013.py
tanganke · Jan 4, 2025 · 7316283 · 7316283
2 parents 9c026f2 + 7f3c168
commit 7316283
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/fusion_bench/dataset/fer2013.py b/fusion_bench/dataset/fer2013.py
@@ -0,0 +1,12 @@
+from datasets import load_dataset
+
+
+def load_fer2013(path: str = "clip-benchmark/wds_fer2013", split: str = "train"):
+    dataset = load_dataset(path, split=split)
+    dataset = dataset.remove_columns(["__key__", "__url__"])
+    dataset = dataset.rename_columns({"jpg": "image", "cls": "label"})
+    return dataset
+
+if __name__ == "__main__":
+    dataset = load_fer2013(split="test")
+    print(dataset)