Merge branch 'master' of https://git.miem.hse.ru/akaldarov/aimm2024

АлдаровАК_БИВ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": 1,
-      "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": "5f793442-2a13-4fae-e8a1-b4654dac0389"
-      },
-      "execution_count": 2,
-      "outputs": [
-        {
-          "output_type": "stream",
-          "name": "stdout",
-          "text": [
-            "Mounted at /content/drive\n"
-          ]
-        }
-      ]
-    },
-    {
-      "cell_type": "code",
-      "source": [
-        "# Path to video\n",
-        "path = '/content/drive/MyDrive/walking_man.mp4'\n",
-        "\n",
-        "# Capturing video\n",
-        "cap = cv2.VideoCapture(path)\n",
-        "\n",
-        "# Creating video writer\n",
-        "writer = cv2.VideoWriter('walking_man_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": 46,
-      "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": 47,
-      "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=2, \\\n",
-        "                        markerType=cv2.MARKER_TILTED_CROSS, line_type=cv2.LINE_AA, \\\n",
-        "                        markerSize=10)"
-      ],
-      "metadata": {
-        "id": "uBNLkUpK_K_U"
-      },
-      "execution_count": 48,
-      "outputs": []
-    },
-    {
-      "cell_type": "code",
-      "source": [
-        "high_color = (255,160,80)\n",
-        "lower_color = (0, 0, 0)"
-      ],
-      "metadata": {
-        "id": "8UMhTvFQ395R"
-      },
-      "execution_count": 49,
-      "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",
-        "\n",
-        "    if ret:\n",
-        "        frame_hsv = cv2.cvtColor(frame, 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()"
-      ],
-      "metadata": {
-        "id": "rpr4YJZN4R0s"
-      },
-      "execution_count": 50,
-      "outputs": []
-    }
-  ]
-}
\ No newline at end of file