add example toy data

ilaria-manco · May 9, 2022 · 7ccd1a7 · 7ccd1a7
1 parent 8736ee4
commit 7ccd1a7
Show file tree

Hide file tree

Showing 6 changed files with 51 additions and 0 deletions.
diff --git a/data/datasets/audiocaption/audio/track_1.npy b/data/datasets/audiocaption/audio/track_1.npy
diff --git a/data/datasets/audiocaption/audio/track_2.npy b/data/datasets/audiocaption/audio/track_2.npy
diff --git a/data/datasets/audiocaption/audio/track_3.npy b/data/datasets/audiocaption/audio/track_3.npy
diff --git a/data/datasets/audiocaption/dataset_test.json b/data/datasets/audiocaption/dataset_test.json
@@ -0,0 +1,17 @@
+[
+    {
+        "audio_id": 1,
+        "caption": "First caption that describes this amazing track!",
+        "audio_path": "track_1.npy"
+    },
+    {
+        "audio_id": 2,
+        "caption": "Second caption that describes this amazing track!",
+        "audio_path": "track_2.npy"
+    },
+    {
+        "audio_id": 3,
+        "caption": "Third caption that describes this amazing track!",
+        "audio_path": "track_3.npy"
+    }
+]
diff --git a/data/datasets/audiocaption/dataset_train.json b/data/datasets/audiocaption/dataset_train.json
@@ -0,0 +1,17 @@
+[
+    {
+        "audio_id": 1,
+        "caption": "First caption that describes this amazing track!",
+        "audio_path": "track_1.npy"
+    },
+    {
+        "audio_id": 2,
+        "caption": "Second caption that describes this amazing track!",
+        "audio_path": "track_2.npy"
+    },
+    {
+        "audio_id": 3,
+        "caption": "Third caption that describes this amazing track!",
+        "audio_path": "track_3.npy"
+    }
+]
diff --git a/data/datasets/audiocaption/dataset_val.json b/data/datasets/audiocaption/dataset_val.json
@@ -0,0 +1,17 @@
+[
+    {
+        "audio_id": 1,
+        "caption": "First caption that describes this amazing track!",
+        "audio_path": "track_1.npy"
+    },
+    {
+        "audio_id": 2,
+        "caption": "Second caption that describes this amazing track!",
+        "audio_path": "track_2.npy"
+    },
+    {
+        "audio_id": 3,
+        "caption": "Third caption that describes this amazing track!",
+        "audio_path": "track_3.npy"
+    }
+]