Load each directory's images and label

Author: Sayed-Hossein-Hosseini

Audio_Classification_Using_CNNs.ipynb

@@ -98,7 +98,7 @@
     {
       "cell_type": "markdown",
       "source": [
-        "## Generate Spectrogram\n",
+        "## **Generate Spectrogram**\n",
         "\n"
       ],
       "metadata": {
@@ -255,19 +255,26 @@
     {
       "cell_type": "code",
       "source": [
-        "## load each directory's images and label:\n",
-        "'''\n",
-        "suggested classes for each directory:\n",
-        "background -> 0\n",
-        "chainsaw -> 1\n",
-        "engine -> 2\n",
-        "storm -> 3\n",
-        "'''\n",
-        "## after loading the images, concat them to the x and y lists respectively\n",
-        "## show some of them\n",
-        "##############################################\n",
-        "############# YOUR CODES GO HERE #############\n",
-        "##############################################"
+        "## Load each directory's images and label\n",
+        "\n",
+        "class_mapping = {\n",
+        "    'background': 0,\n",
+        "    'chainsaw': 1,\n",
+        "    'engine': 2,\n",
+        "    'storm': 3\n",
+        "}\n",
+        "\n",
+        "spectrogram_root = 'dataset/spectrograms'\n",
+        "\n",
+        "for class_name, label in class_mapping.items():\n",
+        "    class_path = os.path.join(spectrogram_root, class_name)\n",
+        "    images, labels = loading_spectrograms(class_path, label)\n",
+        "\n",
+        "    x.extend(images)\n",
+        "    y.extend(labels)\n",
+        "\n",
+        "# Show a sample of loaded images\n",
+        "show_images(x)"
       ],
       "metadata": {
         "id": "qV6bm8VtMeTS"
@@ -312,7 +319,7 @@
     {
       "cell_type": "markdown",
       "source": [
-        "## Build and train a CNN"
+        "## **Build and train a CNN**"
       ],
       "metadata": {
         "id": "LVM1AzW2R3bz"