Upload ПР1 .ipynb

АлдаровАК_БИВ211_ПР1.ipynb

+{
+  "nbformat": 4,
+  "nbformat_minor": 0,
+  "metadata": {
+    "colab": {
+      "provenance": []
+    },
+    "kernelspec": {
+      "name": "python3",
+      "display_name": "Python 3"
+    },
+    "language_info": {
+      "name": "python"
+    }
+  },
+  "cells": [
+    {
+      "cell_type": "code",
+      "execution_count": 129,
+      "metadata": {
+        "id": "rkL66VFAv45A"
+      },
+      "outputs": [],
+      "source": [
+        "import cv2\n",
+        "import numpy as np\n",
+        "import matplotlib.pyplot as plt"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "from google.colab import drive\n",
+        "drive.mount('/content/drive')"
+      ],
+      "metadata": {
+        "id": "Rr1YMcU711VQ",
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        },
+        "outputId": "2fa59ab5-c20d-46d8-a976-4bb3002ac0cd"
+      },
+      "execution_count": 130,
+      "outputs": [
+        {
+          "output_type": "stream",
+          "name": "stdout",
+          "text": [
+            "Drive already mounted at /content/drive; to attempt to forcibly remount, call drive.mount(\"/content/drive\", force_remount=True).\n"
+          ]
+        }
+      ]
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "# Path to video\n",
+        "path = '/content/drive/MyDrive/prybakov_walks.mp4'\n",
+        "\n",
+        "# Capturing video\n",
+        "cap = cv2.VideoCapture(path)\n",
+        "\n",
+        "# Creating video writer\n",
+        "writer = cv2.VideoWriter('prybakov_walks_result.mp4', # video name\n",
+        "                         cv2.VideoWriter_fourcc(*'MP4V'), # codec\n",
+        "                         25, # FPS\n",
+        "                         (int(cap.get(3)),int(cap.get(4))) # dimentions X:Y\n",
+        "                        )"
+      ],
+      "metadata": {
+        "id": "9b9vy_GWwtDE"
+      },
+      "execution_count": 131,
+      "outputs": []
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "# Function to find the center of the biggest blob of colored pixels on image\n",
+        "def find_centroid(img):\n",
+        "    contours, _ = cv2.findContours(img, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE) # Finding blobs\n",
+        "    if len(contours) > 0:\n",
+        "        largest_contour = max(contours, key=cv2.contourArea) # Finding the biggest blob\n",
+        "\n",
+        "        moments = cv2.moments(largest_contour) # Finding contour moments\n",
+        "\n",
+        "        if moments[\"m00\"] != 0:\n",
+        "            cx = int(moments[\"m10\"] / moments[\"m00\"])\n",
+        "            cy = int(moments[\"m01\"] / moments[\"m00\"])\n",
+        "        else:\n",
+        "            cx, cy = 0, 0\n",
+        "\n",
+        "        return (cx, cy)"
+      ],
+      "metadata": {
+        "id": "NcGSNZoG8sSw"
+      },
+      "execution_count": 132,
+      "outputs": []
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "# Function to draw crosses from all previous frames\n",
+        "def draw_cross(frame, points, color=(150,200,255)):\n",
+        "    for cx, cy in points:\n",
+        "        cv2.drawMarker(frame, (cx, cy), color=color, thickness=1, \\\n",
+        "                        markerType=cv2.MARKER_TILTED_CROSS, line_type=cv2.LINE_AA, \\\n",
+        "                        markerSize=25)"
+      ],
+      "metadata": {
+        "id": "uBNLkUpK_K_U"
+      },
+      "execution_count": 139,
+      "outputs": []
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "lower_color = (160,0,25)\n",
+        "high_color = (200, 255, 255)"
+      ],
+      "metadata": {
+        "id": "nakl8vS_C0zM"
+      },
+      "execution_count": 134,
+      "outputs": []
+    },
+    {
+      "cell_type": "code",
+      "source": [
+        "cap.set(cv2.CAP_PROP_POS_FRAMES, 0)\n",
+        "\n",
+        "ret = True\n",
+        "\n",
+        "# List of coordinates of all centroids\n",
+        "points = []\n",
+        "\n",
+        "while ret:\n",
+        "    ret, frame = cap.read()\n",
+        "    if ret:\n",
+        "        frame_temp = frame.copy() # Copying the image for manipulations\n",
+        "\n",
+        "        frame_hsv = cv2.cvtColor(frame_temp, cv2.COLOR_BGR2HSV) # BGR -> HSV\n",
+        "        frame_thres = cv2.inRange(frame_hsv, lower_color, high_color) # Image thresholding\n",
+        "\n",
+        "        coordinates = find_centroid(frame_thres)\n",
+        "        if coordinates is not None:\n",
+        "            points.append(coordinates)\n",
+        "\n",
+        "        draw_cross(frame, points)\n",
+        "\n",
+        "        writer.write(frame)\n",
+        "\n",
+        "writer.release()\n",
+        "print('Done!')"
+      ],
+      "metadata": {
+        "colab": {
+          "base_uri": "https://localhost:8080/"
+        },
+        "id": "rpr4YJZN4R0s",
+        "outputId": "651f0cab-89c8-4438-ea2b-e5f3e2b7ddc1"
+      },
+      "execution_count": 140,
+      "outputs": [
+        {
+          "output_type": "stream",
+          "name": "stdout",
+          "text": [
+            "Done!\n"
+          ]
+        }
+      ]
+    }
+  ]
+}
\ No newline at end of file