Added mode feedback

pythonioncoder · pythonioncoder · commit 66b5edbe9250 · 2024-06-29T12:15:09.000-04:00
diff --git a/Prototype_2_Electric_Boogalooo.ipynb b/Prototype_2_Electric_Boogalooo.ipynb
@@ -5,7 +5,7 @@
     "colab": {
       "provenance": [],
       "mount_file_id": "1YJfEK8zha4PfUk2yP9y-XbO4vtmn83gE",
-      "authorship_tag": "ABX9TyPqC8aZOBxrn5vwvNxHoub4",
+      "authorship_tag": "ABX9TyPf94t3YgjtLkVRDYA222aU",
       "include_colab_link": true
     },
     "kernelspec": {
@@ -107,7 +107,8 @@
         "import os\n",
         "import glob\n",
         "import csv\n",
-        "import warnings"
+        "import warnings\n",
+        "from statistics import mode"
       ],
       "metadata": {
         "id": "whZwvEdKIksm"
@@ -278,7 +279,7 @@
       "metadata": {
         "id": "RHyJwan8acCz"
       },
-      "execution_count": null,
+      "execution_count": 17,
       "outputs": []
     },
     {
@@ -297,18 +298,18 @@
       ],
       "metadata": {
         "id": "Zng2DvUJcJjc",
-        "outputId": "5e773745-adb7-45d0-fbf6-dc47ad96e633",
+        "outputId": "06119dd8-3a96-4a01-ef98-a102e9126ca7",
         "colab": {
           "base_uri": "https://localhost:8080/"
         }
       },
-      "execution_count": null,
+      "execution_count": 18,
       "outputs": [
         {
           "name": "stdout",
           "output_type": "stream",
           "text": [
-            "What model to use? heavy\n"
+            "What model to use? (lite/heavy)heavy\n"
           ]
         }
       ]
@@ -332,10 +333,23 @@
         "  vid_path = root_path + '/Fit Form AI Resources/Videos/'"
       ],
       "metadata": {
-        "id": "brFyKQV5aEuK"
+        "id": "brFyKQV5aEuK",
+        "outputId": "29211e6f-c1d4-4e2c-dffe-618c5ebbab0e",
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        }
       },
-      "execution_count": null,
-      "outputs": []
+      "execution_count": 19,
+      "outputs": [
+        {
+          "name": "stdout",
+          "output_type": "stream",
+          "text": [
+            "Would you like to record and save the video you create below? (Y / N)\n",
+            "Y\n"
+          ]
+        }
+      ]
     },
     {
       "cell_type": "code",
@@ -349,7 +363,7 @@
         "if record:\n",
         "  out_video = cv.VideoWriter(vid_path + time.asctime(time.localtime()) + '.mp4',cv.VideoWriter_fourcc(*'DIVX'), 24, (640, 480))\n",
         "landmarks_list = []\n",
-        "feedback_list = [\"\"]\n",
+        "feedback_list = []\n",
         "selected_exercise = input(\"What exercise are you tracking?\\n\")\n",
         "\n",
         "\n",
@@ -371,9 +385,8 @@
         "    # print(\"Frame\", timestamp_ms)\n",
         "    working_dstruct = lm_dstruct(landmarks_list[-1])\n",
         "    feedback_temp = working_dstruct.feedback(selected_exercise)\n",
-        "    if feedback_list[-1] != feedback_temp:\n",
-        "      print(feedback_temp)\n",
         "    feedback_list.append(feedback_temp)\n",
+        "    print(mode(feedback_list[-1:-4:-1]) if len(feedback_list) > 2 else feedback_temp)\n",
         "    if record:\n",
         "      out_video.write(annotated_image)\n",
         "\n",
@@ -535,4 +548,4 @@
       "outputs": []
     }
   ]
-}
+}
