python提取视频流并截图

当我们成功获取视频流后，可能需要获取实时的截图，接下来介绍三种读取视频流并自动进行截图的方法。

OpenCV 直接读取 RTSP 流并截图

import cv2

# RTSP流地址
rtsp_url = 'your_rtsp'

# 创建视频捕捉对象，使用FFmpeg
cap = cv2.VideoCapture(rtsp_url, cv2.CAP_FFMPEG)

# 检查视频捕捉对象是否打开成功
if not cap.isOpened():
    print("Error: Could not open video stream")
else:
    # 读取一帧
    ret, frame = cap.read()

    if ret:
        # 保存截图
        cv2.imwrite('screenshot.jpg', frame)
        print("Screenshot saved successfully")
    else:
        print("Error: Could not read frame")

# 释放视频捕捉对象
cap.release()

FFmpeg 提取视频流并处理截图

import ffmpeg
import cv2
import numpy as np

# RTSP流地址
rtsp_url = 'your_rtsp'

# 使用FFmpeg从RTSP流中读取一帧
out, _ = (
    ffmpeg
    .input(rtsp_url)
    .output('pipe:', vframes=1, format='image2', vcodec='mjpeg')
    .run(capture_stdout=True, capture_stderr=True)
)

# 将输出转换为numpy数组
nparr = np.frombuffer(out, np.uint8)
# 解码图像
frame = cv2.imdecode(nparr, cv2.IMREAD_COLOR)

# 保存截图
cv2.imwrite('screenshot.jpg', frame)
print("Screenshot saved successfully")

用 VLC 的 Python 接口来读取 RTSP 流

import vlc
import time
import cv2
import numpy as np

# RTSP流地址
rtsp_url = 'your_rtsp'

# 创建VLC实例
instance = vlc.Instance()
player = instance.media_player_new()
media = instance.media_new(rtsp_url)
media.add_option('network-caching=300')
player.set_media(media)

# 播放视频
player.play()
time.sleep(5)  # 等待视频加载

# 获取视频帧
ret, frame = player.video_take_snapshot(0, 'screenshot.jpg', 0, 0)
if ret == 0:
    print("Screenshot saved successfully")
else:
    print("Error: Could not read frame")

首先，确保 VLC 已安装。如果还没有安装，可以从 VLC 官方网站下载并安装。
然后，将 VLC 的安装目录（包含libvlc.dll 文件的目录）添加到系统的环境变量 PATH 中。以下是在 Windows 10 上添加环境变量的方法：

• 右键点击“此电脑”，选择“属性”。
• 选择“高级系统设置”，点击“环境变量”。
• 在“系统变量”部分，找到并选择“Path”，然后点击“编辑”。
• 点击“新建”，并添加 VLC 安装目录路径，例如 C:\Program Files\VideoLAN\VLC。 点击“确定”保存更改。