CV-视频基本处理

OpenCV 提供了丰富的功能来处理视频，包括读取、显示、保存和处理视频帧。下面是一些常见的视频基本处理操作的示例代码，包括视频的读取、显示、保存以及帧的基本处理。

1. 读取和显示视频

示例代码：

import cv2

# 打开视频文件或摄像头
video_capture = cv2.VideoCapture('video.mp4')  # 或者使用 0 作为参数来打开摄像头

while True:
    # 读取视频帧
    ret, frame = video_capture.read()
    
    # 检查是否成功读取帧
    if not ret:
        break
    
    # 显示帧
    cv2.imshow('Video', frame)
    
    # 按 'q' 键退出
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

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

2. 保存视频

示例代码：

import cv2

# 打开视频文件或摄像头
video_capture = cv2.VideoCapture('video.mp4')

# 获取视频的基本属性
fps = video_capture.get(cv2.CAP_PROP_FPS)
width = int(video_capture.get(cv2.CAP_PROP_FRAME_WIDTH))
height = int(video_capture.get(cv2.CAP_PROP_FRAME_HEIGHT))

# 定义输出视频的编码方式和输出文件
fourcc = cv2.VideoWriter_fourcc(*'XVID')
output_video = cv2.VideoWriter('output_video.avi', fourcc, fps, (width, height))

while True:
    # 读取视频帧
    ret, frame = video_capture.read()
    
    # 检查是否成功读取帧
    if not ret:
        break
    
    # 写入帧到输出视频
    output_video.write(frame)

# 释放资源
video_capture.release()
output_video.release()
cv2.destroyAllWindows()

3. 对视频帧进行基本处理（如灰度化）

示例代码：

import cv2

# 打开视频文件或摄像头
video_capture = cv2.VideoCapture('video.mp4')

# 获取视频的基本属性
fps = video_capture.get(cv2.CAP_PROP_FPS)
width = int(video_capture.get(cv2.CAP_PROP_FRAME_WIDTH))
height = int(video_capture.get(cv2.CAP_PROP_FRAME_HEIGHT))

# 定义输出视频的编码方式和输出文件
fourcc = cv2.VideoWriter_fourcc(*'XVID')
output_video = cv2.VideoWriter('processed_video.avi', fourcc, fps, (width, height))

while True:
    # 读取视频帧
    ret, frame = video_capture.read()
    
    # 检查是否成功读取帧
    if not ret:
        break
    
    # 转换为灰度图像
    gray_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    
    # 将灰度图像转换回 BGR 以保存为视频
    gray_frame_bgr = cv2.cvtColor(gray_frame, cv2.COLOR_GRAY2BGR)
    
    # 显示帧
    cv2.imshow('Video', gray_frame_bgr)
    
    # 写入处理后的帧到输出视频
    output_video.write(gray_frame_bgr)
    
    # 按 'q' 键退出
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

# 释放资源
video_capture.release()
output_video.release()
cv2.destroyAllWindows()

4. 从视频中提取帧

示例代码：

import cv2

# 打开视频文件或摄像头
video_capture = cv2.VideoCapture('video.mp4')

frame_count = 0

while True:
    # 读取视频帧
    ret, frame = video_capture.read()
    
    # 检查是否成功读取帧
    if not ret:
        break
    
    # 保存每 30 帧为一个图像文件
    if frame_count % 30 == 0:
        cv2.imwrite(f'frame_{frame_count}.jpg', frame)
    
    frame_count += 1

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

代码解释

1. 读取和显示视频：

   • cv2.VideoCapture()：打开视频文件或摄像头。
   • cv2.imshow()：显示视频帧。
   • cv2.waitKey()：设置显示帧的延迟，并检查用户是否按下了退出键。

2. 保存视频：

   • cv2.VideoWriter()：定义输出视频文件和编码方式。
   • output_video.write()：将处理后的帧写入输出视频文件。

3. 对视频帧进行处理：

   • 使用 cv2.cvtColor() 将视频帧转换为灰度图像，然后再转换回 BGR 以便保存。

4. 提取帧：

   • 使用 cv2.imwrite() 保存每隔一定帧数的图像到文件。

注意事项

• 文件路径：确保视频文件路径正确。
• 编解码器：选择合适的视频编解码器（如 XVID、MJPG、MP4V）。
• 性能：处理高清视频时可能需要更多的计算资源和内存。

这些基本操作可以帮助你在 OpenCV 中处理视频数据，实现各种视频处理和分析任务。