Add data loading example for anechoic data to mixture generator notebook

TCord · May 17, 2022 · 91d3b60 · 91d3b60
1 parent dc48a17
commit 91d3b60
Showing 1 changed file with 21 additions and 1 deletion.
diff --git a/notebooks/mixture_generator.ipynb b/notebooks/mixture_generator.ipynb
@@ -153,6 +153,26 @@
     "plot_mixtures(ds)"
    ]
   },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "ff92352e",
+   "metadata": {
+    "pycharm": {
+     "name": "#%%\n"
+    }
+   },
+   "outputs": [],
+   "source": [
+    "# Load audio\n",
+    "from mms_msg import keys\n",
+    "ds = ds\\\n",
+    "    .map(lambda example: load_audio(example, keys.ORIGINAL_SOURCE))\\\n",
+    "    .map(mms_msg.simulation.anechoic.anechoic_scenario_map_fn)\n",
+    "ex = ds[0]\n",
+    "pb.io.play(ex['audio_data']['observation'], sample_rate=8000)"
+   ]
+  },
   {
    "cell_type": "markdown",
    "id": "0dd6c614",
@@ -434,4 +454,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 5
-}
+}