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()