获取视频宽度和高度 cv2 - Python

在使用OpenCV处理视频时，通常需要获取视频的宽度和高度，以便进行后续的操作。本文将介绍Python中使用cv2获取视频宽度和高度的方法。

安装OpenCV

在Python中使用cv2需要先安装OpenCV库。可以通过以下命令安装：

pip install opencv-python

加载视频

使用cv2可以轻松地从文件中加载视频，示例代码如下：

import cv2

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

# 循环读取视频帧
while True:
    ret, frame = cap.read()
    
    if ret:
        # 处理视频帧
        pass
        
        # 显示视频帧
        cv2.imshow('frame', frame)
        
        # 检测按键事件
        if cv2.waitKey(25) & 0xFF == ord('q'):
            break
    else:
        break

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

在循环中，使用cap.read()读取视频帧。如果成功读取到视频帧，ret为True，frame为当前帧的图像数据；否则都为False。

获取视频宽度和高度

在加载视频之后，可以使用cap.get(cv2.CAP_PROP_FRAME_WIDTH)和cap.get(cv2.CAP_PROP_FRAME_HEIGHT)获取视频的宽度和高度。

import cv2

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

# 获取视频宽度和高度
width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
print('Video size: {}x{}'.format(width, height))

# 循环读取视频帧
while True:
    ret, frame = cap.read()
    
    if ret:
        # 处理视频帧
        pass
        
        # 显示视频帧
        cv2.imshow('frame', frame)
        
        # 检测按键事件
        if cv2.waitKey(25) & 0xFF == ord('q'):
            break
    else:
        break

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

在上述代码中，我们使用int()函数将获取到的浮点数转换为整数类型。

结论

通过上述介绍，相信大家已经学会了如何使用cv2获取视频宽度和高度。在处理视频时，掌握这些基础知识是非常重要的。