Python读取视频的两种方法(imageio和cv2)

Python读取视频有很多方法，其中比较常用的两种方法分别是：imageio和cv2。下面将分别详细讲解这两种方法的使用步骤和示例代码。

Python读取视频有很多方法，其中比较常用的两种方法分别是：imageio和cv2。下面将分别详细讲解这两种方法的使用步骤和示例代码。

一、使用imageio读取视频

1. 安装imageio

使用pip命令安装imageio：

pip install imageio

2. 读取视频文件

读取视频文件需要用到imageio的imread函数。示例代码如下：

import imageio

# 打开视频文件，read_video函数会返回一个生成器对象
reader = imageio.get_reader('test.mp4')

# 遍历生成器对象并获取每一帧图片
for i, im in enumerate(reader):
    # 对图片做处理
    im_processed = process_image(im)
    # 显示处理后的图片
    imshow(im_processed)

生成器对象reader可以迭代取出视频的每一帧。imread函数支持读取多种视频格式，如mp4、avi、mpeg等等。

3. 关闭视频文件

读取视频文件后，一定要关闭它，以释放资源。可以使用reader.close()方法，也可以使用with语句来关闭文件：

import imageio

with imageio.get_reader('test.mp4') as reader:
    for i, im in enumerate(reader):
        # 对图片做处理
        im_processed = process_image(im)
        # 显示处理后的图片
        imshow(im_processed)

二、使用cv2读取视频

1. 安装OpenCV和numpy

使用pip命令安装OpenCV和numpy：

pip install opencv-python numpy

2. 读取视频文件

使用cv2模块的VideoCapture类可以读取视频文件。示例代码如下：

import cv2

# 打开视频文件
cap = cv2.VideoCapture('test.mp4')

# 读取视频直到结束
while(cap.isOpened()):
    # 读取视频的下一帧
    ret, frame = cap.read()
    if ret:
        # 对图片做处理
        frame_processed = process_frame(frame)
        # 显示处理后的图片
        cv2.imshow('frame processed', frame_processed)
    else:
        break

    # 等待用户按下q键退出
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

# 释放资源
cap.release()
cv2.destroyAllWindows()

VideoCapture类的构造函数可以传入视频文件名或设备编号，如cap = cv2.VideoCapture(0)打开摄像头。

3. 关闭视频文件

读取视频文件后，一定要关闭它，以释放资源。可以使用cap.release()方法来关闭文件。

示例1：读取本地视频文件

import imageio

# 打开视频文件，read_video函数会返回一个生成器对象
reader = imageio.get_reader('test.mp4')

# 遍历生成器对象并获取每一帧图片
for i, im in enumerate(reader):
    # 对图片做处理
    im_processed = process_image(im)
    # 显示处理后的图片
    imshow(im_processed)

示例2：读取网络摄像头实时视频流

import cv2

# 打开视频流
cap = cv2.VideoCapture('http://ip:port/video')

# 读取视频流直到结束
while True:
    # 读取视频的下一帧
    ret, frame = cap.read()
    if ret:
        # 对图片做处理
        frame_processed = process_frame(frame)
        # 显示处理后的图片
        cv2.imshow('frame processed', frame_processed)
    else:
        break

    # 等待用户按下q键退出
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

# 释放资源
cap.release()
cv2.destroyAllWindows()

这个示例中，使用cv2的VideoCapture类读取了一个网络摄像头的实时视频流，流的URL地址为'http://ip:port/video'。注意，有些网络摄像头需要先输入账号密码。