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

6c856209 · Алдаров Артем Константинович · a713764f · 6c856209
Commit 6c856209 authored 5 months ago by Алдаров Артем Константинович
Hide whitespace changes
Inline Side-by-side

Showing

with 19 additions and 33 deletions
+19 -33
--- a/АлдаровАК_БИВ211_ПР1.ipynb
+++ b/АлдаровАК_БИВ211_ПР1.ipynb
@@ -16,7 +16,7 @@
  "cells": [
    {
      "cell_type": "code",
-      "execution_count": 72,
+      "execution_count": 1,
      "metadata": {
        "id": "rkL66VFAv45A"
      },
@@ -38,15 +38,15 @@
        "colab": {
          "base_uri": "https://localhost:8080/"
        },
-        "outputId": "ada8ee6d-c7f7-4d51-e19f-de5e6fbc299a"
+        "outputId": "5f793442-2a13-4fae-e8a1-b4654dac0389"
      },
-      "execution_count": 73,
+      "execution_count": 2,
      "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"
+            "Mounted at /content/drive\n"
          ]
        }
      ]
@@ -70,7 +70,7 @@
      "metadata": {
        "id": "9b9vy_GWwtDE"
      },
-      "execution_count": 74,
+      "execution_count": 46,
      "outputs": []
    },
    {
@@ -95,7 +95,7 @@
      "metadata": {
        "id": "NcGSNZoG8sSw"
      },
-      "execution_count": 75,
+      "execution_count": 47,
      "outputs": []
    },
    {
@@ -106,24 +106,24 @@
        "    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=15)"
+        "                        markerSize=10)"
      ],
      "metadata": {
        "id": "uBNLkUpK_K_U"
      },
-      "execution_count": 76,
+      "execution_count": 48,
      "outputs": []
    },
    {
      "cell_type": "code",
      "source": [
-        "high_color = (87,86,104)\n",
-        "lower_color = (35, 30, 40)"
+        "high_color = (255,160,80)\n",
+        "lower_color = (0, 0, 0)"
      ],
      "metadata": {
-        "id": "nakl8vS_C0zM"
+        "id": "8UMhTvFQ395R"
      },
-      "execution_count": 77,
+      "execution_count": 49,
      "outputs": []
    },
    {
@@ -140,8 +140,8 @@
        "    ret, frame = cap.read()\n",
        "\n",
        "    if ret:\n",
-        "        frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB) # BGR -> RGB\n",
-        "        frame_thres = cv2.inRange(frame, lower_color, high_color) # Image thresholding\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",
@@ -149,30 +149,15 @@
        "\n",
        "        draw_cross(frame, points)\n",
        "\n",
-        "        frame = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR)\n",
-        "\n",
        "        writer.write(frame)\n",
        "\n",
-        "writer.release()\n",
-        "print('Done!')"
+        "writer.release()"
      ],
      "metadata": {
-        "colab": {
-          "base_uri": "https://localhost:8080/"
-        },
-        "id": "rpr4YJZN4R0s",
-        "outputId": "48afa7f2-0ac3-4880-cf7e-5ba51d2d7eaf"
+        "id": "rpr4YJZN4R0s"
      },
-      "execution_count": 78,
-      "outputs": [
-        {
-          "output_type": "stream",
-          "name": "stdout",
-          "text": [
-            "Done!\n"
-          ]
-        }
-      ]
+      "execution_count": 50,
+      "outputs": []
    }
  ]
 }
\ No newline at end of file
 %% Cell type:code id: tags:

 ``` 
 import cv2
 import numpy as np
 import matplotlib.pyplot as plt
 ```

 %% Cell type:code id: tags:

 ``` 
 from google.colab import drive
 drive.mount('/content/drive')
 ```

 %% Output

-    Drive already mounted at /content/drive; to attempt to forcibly remount, call drive.mount("/content/drive", force_remount=True).
+    Mounted at /content/drive

 %% Cell type:code id: tags:

 ``` 
 # Path to video
 path = '/content/drive/MyDrive/walking_man.mp4'

 # Capturing video
 cap = cv2.VideoCapture(path)

 # Creating video writer
 writer = cv2.VideoWriter('walking_man_result.mp4', # video name
                         cv2.VideoWriter_fourcc(*'MP4V'), # codec
                         25, # FPS
                         (int(cap.get(3)),int(cap.get(4))) # dimentions X:Y
                        )
 ```

 %% Cell type:code id: tags:

 ``` 
 # Function to find the center of the biggest blob of colored pixels on image
 def find_centroid(img):
    contours, _ = cv2.findContours(img, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE) # Finding blobs
    if len(contours) > 0:
        largest_contour = max(contours, key=cv2.contourArea) # Finding the biggest blob

        moments = cv2.moments(largest_contour) # Finding contour moments

        if moments["m00"] != 0:
            cx = int(moments["m10"] / moments["m00"])
            cy = int(moments["m01"] / moments["m00"])
        else:
            cx, cy = 0, 0

        return (cx, cy)
 ```

 %% Cell type:code id: tags:

 ``` 
 # Function to draw crosses from all previous frames
 def draw_cross(frame, points, color=(150,200,255)):
    for cx, cy in points:
        cv2.drawMarker(frame, (cx, cy), color=color, thickness=2, \
                        markerType=cv2.MARKER_TILTED_CROSS, line_type=cv2.LINE_AA, \
-                        markerSize=15)
+                        markerSize=10)
 ```

 %% Cell type:code id: tags:

 ``` 
-high_color = (87,86,104)
-lower_color = (35, 30, 40)
+high_color = (255,160,80)
+lower_color = (0, 0, 0)
 ```

 %% Cell type:code id: tags:

 ``` 
 cap.set(cv2.CAP_PROP_POS_FRAMES, 0)

 ret = True

 # List of coordinates of all centroids
 points = []

 while ret:
    ret, frame = cap.read()

    if ret:
-        frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB) # BGR -> RGB
-        frame_thres = cv2.inRange(frame, lower_color, high_color) # Image thresholding
+        frame_hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV) # BGR -> HSV
+        frame_thres = cv2.inRange(frame_hsv, lower_color, high_color) # Image thresholding

        coordinates = find_centroid(frame_thres)
        if coordinates is not None:
            points.append(coordinates)

        draw_cross(frame, points)

-        frame = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR)
-
        writer.write(frame)

 writer.release()
-print('Done!')
 ```
-
-%% Output
-
-    Done!