A股上市公司传智教育(股票代码 003032)旗下技术交流社区北京昌平校区

 找回密码
 加入黑马

QQ登录

只需一步,快速开始

内容来自OpenCV-Python Tutorials 自己翻译整理

目标:

  • 读取视频、显示视频、保存视频
  • 使用摄像头拍摄并显示
  • 学习cv2.VideoCapture(), cv2.VideoWriter()函数

使用摄像头拍摄视频:
(我的用的是笔记本电脑上面的摄像头)
OpenCV 提供了一个非常简单的接口。我们把视频转换成黑白格式,然后显示出来。

首先要创建一个VideoCapture对象。参数可以是设备标示(device index)或是一个视频文件的名称。设备标示是指定使用哪个摄像头的编号。通常一个摄像头会连接上。所以只需要传递参数0或者-1。可以选择第二个摄像头通过传递参数1,以此类推。之后就能通过摄像头获得每一帧的数据,在最后释放摄像头资源。

import numpy as np

import cv2


cap = cv2.VideoCapture(0)


while(True):

    # Capture frame-by-frame

    ret, frame = cap.read()


    # Our operations on the frame come here

    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)


    # Display the resulting frame

    cv2.imshow('frame',gray)

    if cv2.waitKey(1) & 0xFF == ord('q'):

        break


# When everything done, release the capture

cap.release()

cv2.destroyAllWindows()

cap.read() 函数返回一个布尔类型。如果读的视频帧是正确的,返回True,所以可以通过返回值检查。


有时,cap可能没有初始化摄像头,可以检查是否初始化或者使用cap.isOpened() 函数,如果返回True则继续,否则使用cap.open()


可以使用cap.get(proid)函数来设置视频的属性,proid是0到18的数字每个数组表示一个视频的属性,具体见Property Identifier

其中一些属性可以是用cap.set(proid,value)方法修改


例如,使用cap.get(3)和cap.get(4)获得每一帧的宽度和高度,获得的结果是640和480,可以把它修改成320×240

ret = cap.set(3,320) ret = cap.set(4,240)


#!/usr/bin/env python

# coding=utf-8


import cv2

import numpy as np

from matplotlib import pyplot as plt


cap = cv2.VideoCapture(0)

ret = cap.set(3,320)

ret = cap.set(4,240)

while(True):

    ret,frame = cap.read()

    gray = cv2.cvtColor(frame,cv2.COLOR_BGR2GRAY)


    cv2.imshow('frame',gray)

    if cv2.waitKey(1) & 0xFF == ord('q'):

        break


cap.release()


cv2.destroyAllWindows()

(不要把set写到循环里,否则会很卡顿)


播放视频文件:

只需要把VideoCapture里面的摄像头标示改成文件名即可,如果视频很小,播放的太快,可以使用参数cv2.waitKey()函数,设置成25毫秒效果不错


import numpy as np

import cv2


cap = cv2.VideoCapture('vtest.avi')


while(cap.isOpened()):

    ret, frame = cap.read()


    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)


    cv2.imshow('frame',gray)

    if cv2.waitKey(1) & 0xFF == ord('q'):

        break


cap.release()


cv2.destroyAllWindows()

存储视频:

要把摄像机拍摄到的视频按帧存起来可以是用cv2.imwrite()函数


首先建立一个VideoWriter对象,然后指定要保存的文件名,例如output.avi。接下来使用FourCC,接下来传递每秒的帧数和帧的大小参数,最后是isColor标记,如果该标记为真,将得到彩色的图片。


FourCC 是用于指定视频编解码器 4 字节代码。代码可以在fourcc.org上找到,该代码是平台相关的。(本人机器为ubuntu 14.04)


import numpy as np

import cv2


cap = cv2.VideoCapture(0)


# Define the codec and create VideoWriter object

fourcc = cv2.VideoWriter_fourcc(*'XVID')

out = cv2.VideoWriter('output.avi',fourcc, 20.0, (640,480))


while(cap.isOpened()):

    ret, frame = cap.read()

    if ret==True:

        frame = cv2.flip(frame,0)


        # write the flipped frame

        out.write(frame)


        cv2.imshow('frame',frame)

        if cv2.waitKey(1) & 0xFF == ord('q'):

            break

    else:

        break


# Release everything if job is finished

cap.release()

out.release()


cv2.destroyAllWindows()

得到的视频是掉转过来的


6 个回复

倒序浏览
优秀,奈斯
回复 使用道具 举报
回复 使用道具 举报
回复 使用道具 举报
回复 使用道具 举报
回复 使用道具 举报
回复 使用道具 举报
您需要登录后才可以回帖 登录 | 加入黑马