摄像头自定义事件响应保存视频

import cv2
import time

frame_rate = 30  # 每秒30帧
frames = []
cap = cv2.VideoCapture(0)
start_time = time.time()

while True:
    ret, frame = cap.read()

    if ret:
        frames.append(frame)

        cv2.imshow('Video', frame)

        if cv2.waitKey(1) & 0xFF == ord('r'):
            timestamp = int(time.time())
            img_name = f'{timestamp}.jpg'
            cv2.imwrite(img_name, frame)
            print(f'保存图片: {img_name}')

            end_time = time.time()
            elapsed_time = end_time - start_time
            frame_count = int(elapsed_time * frame_rate)  # 计算已经保存的帧数

            start_frame = max(0, frame_count - frame_rate*5)  # 判断起始帧
            end_frame = min(frame_count + frame_rate*5, len(frames))  # 判断结束帧

            video_name = f'video_{timestamp}.avi'
            out = cv2.VideoWriter(video_name, cv2.VideoWriter_fourcc(*'DIVX'), frame_rate, (frame.shape[1], frame.shape[0]))
            for i in range(start_frame, end_frame):
                out.write(frames[i])
                print(i)
            out.release()
            print(f'保存视频: {video_name}')
            frames = []
            # frames = []  # 重置frames
            start_time = time.time()

    if cv2.waitKey(1) & 0xFF == 27:
        break

cap.release()
cv2.destroyAllWindows()