Python 的 OpenCV 库来实现直播录屏功能

Python 的 OpenCV 库来实现直播录屏功能

          

import cv2

# 初始化摄像头
capture = cv2.VideoCapture(0)

# 定义编码器并创建 VideoWriter 对象
fourcc = cv2.VideoWriter_fourcc(*'MJPG')
out = cv2.VideoWriter('output.avi', fourcc, 20.0, (640,480))

while(capture.isOpened()):
    # 读取帧
    ret, frame = capture.read()
    if ret==True:
        # 将帧写入视频文件
        out.write(frame)

        # 显示帧
        cv2.imshow('frame',frame)
        if cv2.waitKey(1) & 0xFF == ord('q'):
            break
    else:
        break

# 释放摄像头和视频文件
capture.release()
out.release()
cv2.destroyAllWindows()

在这个代码中，我们使用了 OpenCV 的 VideoCapture 类来初始化摄像头，然后使用 VideoWriter 类将帧写入视频文件中。你可以通过修改 fourcc 参数和输出文件名来控制视频的编码和文件格式。