【python】opencv 利用电脑摄像头捕获图像并保存

目标

利用笔记本的的内置摄像头录像并且保存。
cv2.VideoCapture()、cv2.VideoWriter()、read()、write()的使用。

示例代码

python  

# -*- coding:utf-8-*-

import numpy as np
import cv2

def video_capture(filePath):
    cap = cv2.VideoCapture(0)
    fps = cap.get(cv2.cv.CV_CAP_PROP_FPS)
    size = (int(cap.get(cv2.cv.CV_CAP_PROP_FRAME_WIDTH)),
            int(cap.get(cv2.cv.CV_CAP_PROP_FRAME_HEIGHT)))
    fourcc = cv2.cv.FOURCC(*'CVID')
    out = cv2.VideoWriter(filePath, fourcc, fps, size)

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

        if ret == True:
            frame = cv2.flip(frame, 0)
            gray = cv2.cvtColor(frame,  cv2.COLOR_BGR2GRAY)
            cv2.imshow('iframe', gray)
            out.write(gray)

            if cv2.waitKey(1) & 0xFF == ord('q'):
                break
        else:
            break
    cap.release()
    out.release()
    cv2.destroyAllWindows()

if __name__ == '__main__':
    filePath = 'D:\output.avi'
    video_capture(filePath)

获取视频

cap = cv2.VideoCapture(0) 打开笔记本的内置摄像头。
cap = cv2.VideoCapture('D:\output.avi') 打开视频文件
** 注意**：若初始化摄像头或者打开视频文件不成功，opencv不会提示你。使用print cap.isOpened()查看，若返回值是True，则表明成功，否则返回值是False。

读入视频

cap.read() 按帧读取视频，它的返回值有两个：ret, frame。其中ret是布尔值，如果读取帧是正确的则返回True，如果文件读取到结尾，它的返回值就为False。frame就是每一帧的图像，是个三维矩阵。

播放视频

cv2.imshow('iframe', gray) 播放视频，第一个参数是视频播放窗口的名称，第二个参数是视频的当前帧。
cv2.waitKey(25) 每一帧的播放时间，毫秒级。

停止捕获视频

本示例中有两种停止捕获视频的方式:

if ret == True:    
  frame = cv2.flip(frame, 0)   
  gray = cv2.cvtColor(frame,  cv2.COLOR_BGR2GRAY)     
  cv2.imshow('iframe', gray)    
  out.write(gray) 
   
  if cv2.waitKey(25) & 0xFF == ord('q'):       
     break
  else:   
     break

通过外部键盘输入

cv2.waitKey(25) & 0xFF == ord('q'):  
    break

在25毫秒内如果键盘输入了“q”,则停止捕获视频;

通过cap.read() 的返回值ret，若ret值为False，则停止捕获视频。这种适合读取视频文件时进行判定，通过摄像头录像则只能通过第一种方式停捕获视频。

视频的一些处理方式

灰度视频 gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
本例中是将摄像头捕获的视频转换为灰色并且保存。
视频旋转 cv2.flip(frame, 0) 第一个参数表示要旋转的视频，第二个参数表示旋转的方向，0表示绕x轴旋转，大于0的数表示绕y轴旋转，小于0的负数表示绕x和y轴旋转。

获取视频的参数信息

使用函数 cap.get(propId) 来获得视频的一些参数信息,使用cap.set(propId, value)设置视频的一些参数信息，propId的值从0到18分别为：

**CV_CAP_PROP_POS_MSEC ** Current position of the video file in milliseconds or video capture timestamp.
CV_CAP_PROP_POS_FRAMES 0-based index of the frame to be decoded/captured next.
CV_CAP_PROP_POS_AVI_RATIO Relative position of the video file: 0 - start of the film, 1 - end of the film.
CV_CAP_PROP_FRAME_WIDTH 视频每一帧的宽。
CV_CAP_PROP_FRAME_HEIGHT 视频每一帧的高。
CV_CAP_PROP_FPS 视频的帧速。
CV_CAP_PROP_FOURCC 4个字符表示的视频编码器格式。
CV_CAP_PROP_FRAME_COUNT 视频的帧数。
CV_CAP_PROP_FORMAT Format of the Mat objects returned byretrieve().
CV_CAP_PROP_MODE Backend-specific value indicating the current capture mode.
CV_CAP_PROP_BRIGHTNESS Brightness of the image (only for cameras).
CV_CAP_PROP_CONTRAST Contrast of the image (only for cameras).
CV_CAP_PROP_SATURATION Saturation of the image (only for cameras).
CV_CAP_PROP_HUE Hue of the image (only for cameras).
CV_CAP_PROP_GAIN Gain of the image (only for cameras).
CV_CAP_PROP_EXPOSURE Exposure (only for cameras).
CV_CAP_PROP_CONVERT_RGB Boolean flags indicating whether images should be converted to RGB.
CV_CAP_PROP_WHITE_BALANCE Currently not supported
CV_CAP_PROP_RECTIFICATION Rectification flag for stereo cameras (note: only supported by DC1394 v 2.x backend currently)
注意：当你查询的属性是VideoCapture()不支持的属性时，返回值为0.

保存视频

out = cv2.VideoWriter(filePath, fourcc, 800, size) 设置输出视频的名称，视频的格式，视频的帧速，视频的大小等。
fourcc = cv2.cv.FOURCC(*'CVID') 设置要保存视频的格式。

释放对象和销毁窗口

    cap.release()
    out.release()
    cv2.destroyAllWindows()

作者：刑素素
链接：https://www.jianshu.com/p/182d83926b45
來源：简书
简书著作权归作者所有，任何形式的转载都请联系作者获得授权并注明出处。

【python】opencv 利用电脑摄像头捕获图像并保存

猜你喜欢