stream example cv2

Author: Koldim2001

examples/example_extra_functions.ipynb

@@ -326,6 +326,144 @@
         "---"
       ]
     },
+    {
+      "cell_type": "markdown",
+      "metadata": {},
+      "source": [
+        "# Video stream processing:"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {},
+      "source": [
+        "Webcam stream segmentation:"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "import cv2\n",
+        "from ultralytics import YOLO\n",
+        "from patched_yolo_infer import visualize_results_usual_yolo_inference\n",
+        "\n",
+        "def process_stream(model, camera_id=0):\n",
+        "    # Open the input video file\n",
+        "    cap = cv2.VideoCapture(camera_id)\n",
+        "\n",
+        "    while True:\n",
+        "        ret, frame = cap.read()\n",
+        "        if not ret:\n",
+        "            break\n",
+        "\n",
+        "        # Apply YOLO model to the frame\n",
+        "        result = visualize_results_usual_yolo_inference(\n",
+        "            frame,\n",
+        "            model,\n",
+        "            imgsz=640,\n",
+        "            conf=0.5,\n",
+        "            iou=0.7,\n",
+        "            segment=True,\n",
+        "            thickness=2,\n",
+        "            show_boxes=False,\n",
+        "            show_class=False,\n",
+        "            show_confidences=False,\n",
+        "            fill_mask=True,\n",
+        "            return_image_array=True,\n",
+        "        )\n",
+        "\n",
+        "        # Display the resulting frame\n",
+        "        cv2.imshow('Processed Frame', result)\n",
+        "        if cv2.waitKey(1) & 0xFF == ord('q'):\n",
+        "            break\n",
+        "\n",
+        "    cap.release()\n",
+        "    cv2.destroyAllWindows()\n",
+        "\n",
+        "# Example usage\n",
+        "model = YOLO(\"yolov8m-seg.pt\")  \n",
+        "process_stream(model)"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {},
+      "source": [
+        "Video file stream detection:"
+      ]
+    },
+    {
+      "cell_type": "code",
+      "execution_count": null,
+      "metadata": {},
+      "outputs": [],
+      "source": [
+        "import cv2\n",
+        "from ultralytics import YOLO\n",
+        "from patched_yolo_infer import visualize_results_usual_yolo_inference\n",
+        "\n",
+        "# Function to process a video file\n",
+        "def process_video(input_video_path, output_video_path, model):\n",
+        "    # Open the input video file\n",
+        "    cap = cv2.VideoCapture(input_video_path)\n",
+        "\n",
+        "    # Get the video's frame width and height\n",
+        "    frame_width = int(cap.get(3))\n",
+        "    frame_height = int(cap.get(4))\n",
+        "\n",
+        "    # Define the codec and create VideoWriter object\n",
+        "    fourcc = cv2.VideoWriter_fourcc(*'mp4v')\n",
+        "    out = cv2.VideoWriter(output_video_path, fourcc, 25.0, (frame_width, frame_height))\n",
+        "\n",
+        "    while True:\n",
+        "        ret, frame = cap.read()\n",
+        "        if not ret:\n",
+        "            break\n",
+        "\n",
+        "        # Apply YOLO model to the frame\n",
+        "        result = visualize_results_usual_yolo_inference(\n",
+        "            frame,\n",
+        "            model,\n",
+        "            imgsz=640,\n",
+        "            conf=0.5,\n",
+        "            iou=0.7,\n",
+        "            thickness=2,\n",
+        "            font_scale=1.0,\n",
+        "            show_boxes=True,\n",
+        "            show_confidences=False,\n",
+        "            return_image_array=True\n",
+        "        )\n",
+        "\n",
+        "        # Write the resulting frame to the output video\n",
+        "        out.write(result)\n",
+        "\n",
+        "        # Display the resulting frame (optional)\n",
+        "        cv2.imshow('Processed Frame', result)\n",
+        "        if cv2.waitKey(1) & 0xFF == ord('q'):\n",
+        "            break\n",
+        "\n",
+        "    # Release everything when done\n",
+        "    cap.release()\n",
+        "    out.release()\n",
+        "    cv2.destroyAllWindows()\n",
+        "\n",
+        "# Example usage\n",
+        "model = YOLO(\"yolov8m.pt\")\n",
+        "input_video_path = 'input.mp4'\n",
+        "output_video_path = 'output.mp4'\n",
+        "process_video(input_video_path, output_video_path, model)"
+      ]
+    },
+    {
+      "cell_type": "markdown",
+      "metadata": {},
+      "source": [
+        "---"
+      ]
+    },
     {
       "cell_type": "markdown",
       "metadata": {
@@ -751,7 +889,7 @@
       "name": "python",
       "nbconvert_exporter": "python",
       "pygments_lexer": "ipython3",
-      "version": "3.10.13"
+      "version": "3.11.8"
     }
   },
   "nbformat": 4,