OpenCV学习2-摄像头和视频文件操作

#用摄像头捕获视频

# -*- coding: utf-8 -*-
"""
用摄像头捕获视频
"""
import numpy as np
import cv2 as cv

#为了获取视频，创建一个 VideoCapture 对象。它参数可以是设备的索引号，或者是一个视频文件。设备索引号就是在
#指定要使用的摄像头。一般的笔记本电脑都有内置摄像头。所以参数就是 0。你可以通过设置成 1 或者其他的来选择别的摄像头。
cap = cv.VideoCapture(0)

#死循环
while(True):
    #摄像头逐帧捕捉，ret是返回状态，frame是视频帧
    ret, frame = cap.read()

    #把图像转换为灰度图像
    #gray = cv.cvtColor(frame, cv.COLOR_BGR2GRAY)

    #显示图像
    cv.imshow('frame',frame)
    #输入q时跳出循环
    if cv.waitKey(1) & 0xFF == ord('q'):
        break

#释放摄像头
cap.release()
#关闭窗口
cv.destroyAllWindows()

#从文件中播放视频

# -*- coding: utf-8 -*-
"""
从文件中播放视频
与从摄像头中捕获一样，你只需要把设备索引号改成视频文件的名字。在播放每一帧时，使用 cv2.waiKey() 
设置适当的持续时间。如果设置的太低视频就会播放的非常快，如果设置的太高就会播放的很慢（你可以使用
这种方法控制视频的播放速度）。通常情况下 25 毫秒就可以了。
"""

import numpy as np
import cv2 as cv

cap = cv.VideoCapture('041.rmvb')

#获得视频帧率
fps = cap.get(cv.CAP_PROP_FPS)
print(fps)
#获得视频分辨率，就是宽度和高度，见https://blog.csdn.net/jgw2008/article/details/79316714
size = (int(cap.get(cv.CAP_PROP_FRAME_WIDTH)),
        int(cap.get(cv.CAP_PROP_FRAME_HEIGHT)))
print(size)
'''有时 cap 可能不能成功的初始化摄像头设备。这种情况下上面的代码会报错。你可以使用 cap.isOpened()，
来检查是否成功初始化了。如果返回值是True，那就没有问题。否则就要使用函数 cap.open()。'''
while(cap.isOpened()):
    '''cap.read() 返回一个布尔值（True/False）。如果帧读取的是正确的，就是 True。所以最后你可以通过
    检查他的返回值来查看视频文件是否已经到了结尾。'''
    ret, frame = cap.read()
    
    #gray = cv.cvtColor(frame, cv.COLOR_BGR2GRAY)
    
    cv.imshow('frame',frame)
    if cv.waitKey(1) & 0xFF == ord('q'):
        break
cap.release()
cv.destroyAllWindows()

#用摄像头捕获视频,并写入文件

# -*- coding: utf-8 -*-
"""
用摄像头捕获视频,并写入文件
"""
import numpy as np
import cv2 as cv

#为了获取视频，你应该创建一个 VideoCapture 对象。他的参数可以是设备的索引号，或者是一个视频文件。设备索引号就是在
#指定要使用的摄像头。一般的笔记本电脑都有内置摄像头。所以参数就是 0。你可以通过设置成 1 或者其他的来选择别的摄像头。
cap = cv.VideoCapture(0)

'''这次我们要创建一个 VideoWriter 的对象。我们应该确定一个输出文件的名字。接下来指定 FourCC 编码。
   播放频率和帧的大小也都需要确定。最后一个是 isColor 标签。如果是 True，每一帧就是彩色图，否则就是灰度图。
   FourCC 就是一个 4 字节码，用来确定视频的编码格式。可用的编码列表可以从fourcc.org查到。这是平台依赖的。'''
fourcc = cv.VideoWriter_fourcc(*'XVID')
out = cv.VideoWriter('output.avi',fourcc, 20.0, (640,480))

#死循环
while(True):
    #摄像头逐帧捕捉，ret是返回状态，frame是视频帧
    ret, frame = cap.read()

    #把图像转换为灰度图像
    #gray = cv.cvtColor(frame, cv.COLOR_BGR2GRAY)

    #显示图像
    cv.imshow('frame',frame)
    #保存为文件
    out.write(frame)
    #输入q时跳出循环
    if cv.waitKey(1) & 0xFF == ord('q'):
        break

#释放摄像头
cap.release()
#关闭窗口
cv.destroyAllWindows()