import cv2
import redis
import time
import base64
import datetime
# 连接Redis数据库
r = redis.StrictRedis(host='localhost', port=6379, db=0)
# 打开摄像头
cap = cv2.VideoCapture(0)
while True:
# 读取摄像头画面
ret, frame = cap.read()
now = datetime.datetime.now()
if ret:
# 将画面转换为字节流
_, buffer = cv2.imencode('.jpg', frame)
frame_bytes = buffer.tobytes()
# 检查Redis中的图像数量
image_count = r.dbsize()
if image_count < 100:
# 将字节流转换为base64编码
frame_base64 = base64.b64encode(frame_bytes).decode('utf-8')
# 以当前时间戳为key,超时时间设置为1小时(单位:秒)
# 获取当前日期和时间
# 获取当前时间的秒数
seconds = time.time()
# 将秒数转换为毫秒数
milliseconds = int(seconds * 1000)
r.setex(str(milliseconds), 3600, frame_base64)
print("已保存一张到Redis中,时间戳为: "+str(now))
# 等待300ms
time.sleep(0.3)
else:
time.sleep(0.5)
print("Redis中图像数量已达到100,不再存储新图像,时间: "+str(now))
# 按下'q'键退出循环
if cv2.waitKey(1) & 0xFF == ord('q'):
break
else:
print("无法读取摄像头图像")
break
# 释放资源
cap.release()
cv2.destroyAllWindows()