emotion_second_run.ipynb

{
  "nbformat": 4,
  "nbformat_minor": 0,
  "metadata": {
    "colab": {
      "name": "emotion_second_run.ipynb",
      "provenance": [],
      "collapsed_sections": [],
      "include_colab_link": true
    },
    "kernelspec": {
      "name": "python3",
      "display_name": "Python 3"
    },
    "language_info": {
      "name": "python"
    }
  },
  "cells": [
    {
      "cell_type": "markdown",
      "metadata": {
        "id": "view-in-github",
        "colab_type": "text"
      },
      "source": [
        "<a href=\"https://colab.research.google.com/github/PranavKathar/DeepVision-Emotion-and-Gesture-Detection/blob/master/emotion_second_run.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>"
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "import pandas as pd\n",
        "import numpy as np \n",
        "import matplotlib.pyplot as plt\n",
        "import os\n",
        "import cv2\n",
        "from tqdm import tqdm\n",
        "\n",
        "\n",
        "data = pd.read_csv(\"/content/drive/MyDrive/emotion_2/fer2013(2).csv\")\n",
        "data.head()\n",
        "\n",
        "labels = []\n",
        "usage = []\n",
        "\n",
        "for i in data[\"emotion\"]:\n",
        "    labels.append(i)\n",
        "    \n",
        "for i in data[\"Usage\"]:\n",
        "    usage.append(i)\n",
        "    \n",
        "print(set(labels))\n",
        "print(set(usage))\n",
        "\n"
      ],
      "metadata": {
        "id": "l3xWhsCCqst6",
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "outputId": "27eb19d1-990c-4529-ef9b-d80ab400e268"
      },
      "execution_count": null,
      "outputs": [
        {
          "output_type": "stream",
          "name": "stdout",
          "text": [
            "{0, 1, 2, 3, 4, 5, 6}\n",
            "{'PrivateTest', 'Training', 'PublicTest'}\n"
          ]
        }
      ]
    },
    {
      "cell_type": "code",
      "execution_count": null,
      "metadata": {
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
        "outputId": "a4c46671-2174-48f2-c39d-896c77b2e9b0",
        "id": "o13A6nEgx84N"
      },
      "outputs": [
        {
          "output_type": "stream",
          "name": "stderr",
          "text": [
            "35887it [05:21, 111.62it/s]\n"
          ]
        }
      ],
      "source": [
        "count0 = 0\n",
        "count1 = 0\n",
        "count2 = 0\n",
        "count3 = 0\n",
        "count4 = 0\n",
        "count5 = 0\n",
        "count6 = 0\n",
        "\n",
        "for i, j, k in tqdm(zip(data[\"emotion\"], data[\"pixels\"], data[\"Usage\"])):\n",
        "    pixel = []\n",
        "    pixels = j.split(' ')\n",
        "    for m in pixels:\n",
        "        value = float(m)\n",
        "        pixel.append(value)\n",
        "    pixel = np.array(pixel)\n",
        "    img = pixel.reshape(48, 48)\n",
        "    \n",
        "    if k == \"Training\":\n",
        "        if not os.path.exists(\"drive/MyDrive/emotion_2/train\"):\n",
        "            os.mkdir(\"drive/MyDrive/emotion_2/train\")\n",
        "        if i == 0:\n",
        "            if not os.path.exists(\"drive/MyDrive/emotion_2/train/Angry\"):\n",
        "                os.mkdir(\"drive/MyDrive/emotion_2/train/Angry\")\n",
        "                path = \"drive/MyDrive/emotion_2/train/Angry/\" + str(count0) + \".jpg\"\n",
        "                cv2.imwrite(path , img)\n",
        "                count0 += 1\n",
        "            else:\n",
        "                path = \"drive/MyDrive/emotion_2/train/Angry/\" + str(count0) + \".jpg\"\n",
        "                cv2.imwrite(path , img)\n",
        "                count0 += 1    \n",
        " \n",
        "        # if i == 1:\n",
        "        #     if not os.path.exists(\"drive/MyDrive/emotion_2/train/Disgust\"):\n",
        "        #         os.mkdir(\"drive/MyDrive/emotion_2/train/Disgust\")\n",
        "        #         path = \"drive/MyDrive/emotion_2/train/Disgust/\" + str(count1) + \".jpg\"\n",
        "        #         cv2.imwrite(path , img)\n",
        "        #         count1 += 1\n",
        "        #     else:\n",
        "        #         path = \"drive/MyDrive/emotion_2/train/Disgust/\" + str(count1) + \".jpg\"\n",
        "        #         cv2.imwrite(path , img)\n",
        "        #         count1 += 1\n",
        "        if i == 2:\n",
        "            if not os.path.exists(\"drive/MyDrive/emotion_2/train/Fear\"):\n",
        "                os.mkdir(\"drive/MyDrive/emotion_2/train/Fear\")\n",
        "                path = \"drive/MyDrive/emotion_2/train/Fear/\" + str(count2) + \".jpg\"\n",
        "                cv2.imwrite(path , img)\n",
        "                count2 += 1\n",
        "            else:\n",
        "                path = \"drive/MyDrive/emotion_2/train/Fear/\" + str(count2) + \".jpg\"\n",
        "                cv2.imwrite(path , img)\n",
        "                count2 += 1\n",
        "        if i == 3:\n",
        "            if not os.path.exists(\"drive/MyDrive/emotion_2/train/Happy\"):\n",
        "                os.mkdir(\"drive/MyDrive/emotion_2/train/Happy\")\n",
        "                path = \"drive/MyDrive/emotion_2/train/Happy/\" + str(count3) + \".jpg\"\n",
        "                cv2.imwrite(path , img)\n",
        "                count3 += 1\n",
        "            else:\n",
        "                path = \"drive/MyDrive/emotion_2/train/Happy/\" + str(count3) + \".jpg\"\n",
        "                cv2.imwrite(path , img)\n",
        "                count3 += 1\n",
        "        if i == 4:\n",
        "            if not os.path.exists(\"drive/MyDrive/emotion_2/train/Sad\"):\n",
        "                os.mkdir(\"drive/MyDrive/emotion_2/train/Sad\")\n",
        "                path = \"drive/MyDrive/emotion_2/train/Sad/\" + str(count4) + \".jpg\"\n",
        "                cv2.imwrite(path , img)\n",
        "                count4 += 1\n",
        "            else:\n",
        "                path = \"drive/MyDrive/emotion_2/train/Sad/\" + str(count4) + \".jpg\"\n",
        "                cv2.imwrite(path , img)\n",
        "                count4 += 1\n",
        "        if i == 5:\n",
        "            if not os.path.exists(\"drive/MyDrive/emotion_2/train/Surprise\"):\n",
        "                os.mkdir(\"drive/MyDrive/emotion_2/train/Surprise\")\n",
        "                path = \"drive/MyDrive/emotion_2/train/Surprise/\" + str(count5) + \".jpg\"\n",
        "                cv2.imwrite(path , img)\n",
        "                count5 += 1\n",
        "            else:\n",
        "                path = \"drive/MyDrive/emotion_2/train/Surprise/\" + str(count5) + \".jpg\"\n",
        "                cv2.imwrite(path , img)\n",
        "                count5 += 1\n",
        "        if i == 6:\n",
        "            if not os.path.exists(\"drive/MyDrive/emotion_2/train/Neutral\"):\n",
        "                os.mkdir(\"drive/MyDrive/emotion_2/train/Neutral\")\n",
        "                path = \"drive/MyDrive/emotion_2/train/Neutral/\" + str(count6) + \".jpg\"\n",
        "                cv2.imwrite(path , img)\n",
        "                count6 += 1\n",
        "            else:\n",
        "                path = \"drive/MyDrive/emotion_2/train/Neutral/\" + str(count6) + \".jpg\"\n",
        "                cv2.imwrite(path , img)\n",
        "                count6 += 1\n",
        "    else:\n",
        "        if not os.path.exists(\"drive/MyDrive/emotion_2/validation\"):\n",
        "            os.mkdir(\"drive/MyDrive/emotion_2/validation\")\n",
        "        if i == 0:\n",
        "            if not os.path.exists(\"drive/MyDrive/emotion_2/validation/Angry\"):\n",
        "                os.mkdir(\"drive/MyDrive/emotion_2/validation/Angry\")\n",
        "                path = \"drive/MyDrive/emotion_2/validation/Angry/\" + str(count0) + \".jpg\"\n",
        "                cv2.imwrite(path , img)\n",
        "                count0 += 1\n",
        "            else:\n",
        "                path = \"drive/MyDrive/emotion_2/validation/Angry/\" + str(count0) + \".jpg\"\n",
        "                cv2.imwrite(path , img)\n",
        "                count0 += 1\n",
        "        # if i == 1:\n",
        "        #     if not os.path.exists(\"drive/MyDrive/emotion_2/validation/Disgust\"):\n",
        "        #         os.mkdir(\"drive/MyDrive/emotion_2/validation/Disgust\")\n",
        "        #         path = \"drive/MyDrive/emotion_2/validation/Disgust/\" + str(count1) + \".jpg\"\n",
        "        #         cv2.imwrite(path , img)\n",
        "        #         count1 += 1\n",
        "        #     else:\n",
        "        #         path = \"drive/MyDrive/emotion_2/validation/Disgust/\" + str(count1) + \".jpg\"\n",
        "        #         cv2.imwrite(path , img)\n",
        "        #         count1 += 1\n",
        "        if i == 2:\n",
        "            if not os.path.exists(\"drive/MyDrive/emotion_2/validation/Fear\"):\n",
        "                os.mkdir(\"drive/MyDrive/emotion_2/validation/Fear\")\n",
        "                path = \"drive/MyDrive/emotion_2/validation/Fear/\" + str(count2) + \".jpg\"\n",
        "                cv2.imwrite(path , img)\n",
        "                count2 += 1\n",
        "            else:\n",
        "                path = \"drive/MyDrive/emotion_2/validation/Fear/\" + str(count2) + \".jpg\"\n",
        "                cv2.imwrite(path , img)\n",
        "                count2 += 1\n",
        "        if i == 3:\n",
        "            if not os.path.exists(\"drive/MyDrive/emotion_2/validation/Happy\"):\n",
        "                os.mkdir(\"drive/MyDrive/emotion_2/validation/Happy\")\n",
        "                path = \"drive/MyDrive/emotion_2/validation/Happy/\" + str(count3) + \".jpg\"\n",
        "                cv2.imwrite(path , img)\n",
        "                count3 += 1\n",
        "            else:\n",
        "                path = \"drive/MyDrive/emotion_2/validation/Happy/\" + str(count3) + \".jpg\"\n",
        "                cv2.imwrite(path , img)\n",
        "                count3 += 1\n",
        "        if i == 4:\n",
        "            if not os.path.exists(\"drive/MyDrive/emotion_2/validation/Sad\"):\n",
        "                os.mkdir(\"drive/MyDrive/emotion_2/validation/Sad\")\n",
        "                path = \"drive/MyDrive/emotion_2/validation/Sad/\" + str(count4) + \".jpg\"\n",
        "                cv2.imwrite(path , img)\n",
        "                count4 += 1\n",
        "            else:\n",
        "                path = \"drive/MyDrive/emotion_2/validation/Sad/\" + str(count4) + \".jpg\"\n",
        "                cv2.imwrite(path , img)\n",
        "                count4 += 1\n",
        "        if i == 5:\n",
        "            if not os.path.exists(\"drive/MyDrive/emotion_2/validation/Surprise\"):\n",
        "                os.mkdir(\"drive/MyDrive/emotion_2/validation/Surprise\")\n",
        "                path = \"drive/MyDrive/emotion_2/validation/Surprise/\" + str(count5) + \".jpg\"\n",
        "                cv2.imwrite(path , img)\n",
        "                count5 += 1\n",
        "            else:\n",
        "                path = \"drive/MyDrive/emotion_2/validation/Surprise/\" + str(count5) + \".jpg\"\n",
        "                cv2.imwrite(path , img)\n",
        "                count5 += 1\n",
        "        if i == 6:\n",
        "            if not os.path.exists(\"drive/MyDrive/emotion_2/validation/Neutral\"):\n",
        "                os.mkdir(\"drive/MyDrive/emotion_2/validation/Neutral\")\n",
        "                path = \"drive/MyDrive/emotion_2/validation/Neutral/\" + str(count6) + \".jpg\"\n",
        "                cv2.imwrite(path , img)\n",
        "                count6 += 1\n",
        "            else:\n",
        "                path = \"drive/MyDrive/emotion_2/validation/Neutral/\" + str(count6) + \".jpg\"\n",
        "                cv2.imwrite(path , img)\n",
        "                count6 += 1\n",
        "\n"
      ]
    },
    {
      "cell_type": "code",
      "source": [
        "import matplotlib.pyplot as plt\n",
        "train_directory = \"drive/MyDrive/emotion_2/train/\"\n",
        "train_dir = os.listdir(train_directory)\n",
        "classes = []\n",
        "\n",
        "for folder in train_dir:\n",
        "    classes.append(folder)\n",
        "\n",
        "train_counts = []\n",
        "\n",
        "for folder in train_dir:\n",
        "    class_path = train_directory + folder + \"/\"\n",
        "    list_train = []\n",
        "    count = 0\n",
        "    for file in os.listdir(class_path):\n",
        "        count +=1\n",
        "    \n",
        "    train_counts.append(count)\n",
        "    \n",
        "train_counts\n",
        "\n",
        "\n",
        "plt.bar(classes, train_counts, width=0.5)\n",
        "plt.title(\"Bar Graph of Train Data\")\n",
        "plt.xlabel(\"Classes\")\n",
        "plt.ylabel(\"Counts\")\n",
        "\n",
        "\n",
        "plt.scatter(classes, train_counts)\n",
        "plt.plot(classes, train_counts, '-o')\n",
        "plt.show()"
      ],
      "metadata": {
        "id": "eAm_RivcqhFC"
      },
      "execution_count": null,
      "outputs": []
    }
  ]
}