Open-CV-Implementations

Author: altruistcoder

README.md

@@ -58,6 +58,10 @@ This is a repository of code files which can be used to perform some basic tasks
 
  Python Code to find & draw Contours present in an image using OpenCV.
 
+* [Recording and Saving Videos](https://github.com/altruistcoder/Open-CV-Implementations/blob/master/recording_video.ipynb):
+
+ Python Code to record and save videos with different types of extensions and video quality using OpenCV.
+
 * [Adding Text to Videos](https://github.com/altruistcoder/Open-CV-Implementations/blob/master/text_on_video.ipynb):
 
  Python Code to draw text on a video using OpenCV.

recording_video.ipynb

@@ -0,0 +1,132 @@
+{
+ "cells": [
+ {
+ "cell_type": "markdown",
+ "metadata": {},
+ "source": [
+ "# Recording Video in OpenCV"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 1,
+ "metadata": {},
+ "outputs": [],
+ "source": [
+ "import numpy as np\n",
+ "import cv2\n",
+ "import os"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": 20,
+ "metadata": {},
+ "outputs": [],
+ "source": [
+ "filename = 'video.avi'\n",
+ "frames_per_second = 24.0\n",
+ "my_res = '720p'\n",
+ "\n",
+ "\n",
+ "def change_res(cap, width, height):\n",
+ " cap.set(3, width)\n",
+ " cap.set(4, height)\n",
+ "\n",
+ "\n",
+ "STD_DIMENSIONS = {\n",
+ " \"480p\": (640, 480),\n",
+ " \"720p\": (1280, 720),\n",
+ " \"1080p\": (1920, 1080),\n",
+ " \"4k\": (3840, 2160),\n",
+ "}\n",
+ "\n",
+ "\n",
+ "def get_dims(cap, res='1080p'):\n",
+ " width, height = STD_DIMENSIONS[\"480p\"]\n",
+ " if res in STD_DIMENSIONS:\n",
+ " width,height = STD_DIMENSIONS[res]\n",
+ " change_res(cap, width, height)\n",
+ " return width, height\n",
+ "\n",
+ "\n",
+ "VIDEO_TYPE = {\n",
+ " 'avi': cv2.VideoWriter_fourcc(*'XVID'),\n",
+ " 'mp4': cv2.VideoWriter_fourcc(*'XVID'),\n",
+ "}\n",
+ "\n",
+ "def get_video_type(filename):\n",
+ " filename, ext = os.path.splitext(filename)\n",
+ " if ext in VIDEO_TYPE:\n",
+ " return VIDEO_TYPE[ext]\n",
+ " return VIDEO_TYPE['avi']\n",
+ "\n",
+ "\n",
+ "\n",
+ "cap = cv2.VideoCapture(0)\n",
+ "out = cv2.VideoWriter(filename, get_video_type(filename), frames_per_second, get_dims(cap, my_res))\n",
+ "\n",
+ "while True:\n",
+ " ret, frame = cap.read()\n",
+ " out.write(frame)\n",
+ " cv2.imshow('frame',frame)\n",
+ " if cv2.waitKey(1) & 0xFF == ord('q'):\n",
+ " break\n",
+ "\n",
+ "\n",
+ "cap.release()\n",
+ "out.release()\n",
+ "cv2.destroyAllWindows()"
+ ]
+ },
+ {
+ "cell_type": "code",
+ "execution_count": null,
+ "metadata": {},
+ "outputs": [],
+ "source": []
+ },
+ {
+ "cell_type": "code",
+ "execution_count": null,
+ "metadata": {},
+ "outputs": [],
+ "source": []
+ },
+ {
+ "cell_type": "code",
+ "execution_count": null,
+ "metadata": {},
+ "outputs": [],
+ "source": []
+ },
+ {
+ "cell_type": "code",
+ "execution_count": null,
+ "metadata": {},
+ "outputs": [],
+ "source": []
+ }
+ ],
+ "metadata": {
+ "kernelspec": {
+ "display_name": "Python 3",
+ "language": "python",
+ "name": "python3"
+ },
+ "language_info": {
+ "codemirror_mode": {
+ "name": "ipython",
+ "version": 3
+ },
+ "file_extension": ".py",
+ "mimetype": "text/x-python",
+ "name": "python",
+ "nbconvert_exporter": "python",
+ "pygments_lexer": "ipython3",
+ "version": "3.5.2"
+ }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}