import cv2

# 初始化摄像头
cap = cv2.VideoCapture(0)  # 0 通常是默认摄像头的标识

# 检查摄像头是否成功打开
if not cap.isOpened():
    print("无法打开摄像头")
    exit()

# 设置拍照参数
# 例如，拍照的帧数
frame_count = 0
# 拍照的最大帧数
max_frames = 1000

# 拍照循环
while True:
    # 读取摄像头的下一帧
    ret, frame = cap.read()

    # 如果正确读取帧，ret为True
    if not ret:
        print("无法接收帧，请退出")
        break

    # 如果达到最大帧数，则保存图片并退出
    if frame_count >= max_frames:
        # 保存图片
        cv2.imwrite('snapshot1.jpg', frame)
        print("图片已保存")
        break

    # 显示当前帧
    cv2.imshow('Frame', frame)

    # 等待键盘输入，按 's' 键进行拍照
    key = cv2.waitKey(1) & 0xFF
    if key == ord('s'):
        # 保存图片
        cv2.imwrite('snapshot2.jpg', frame)
        print("图片已保存")
        break

    # 计数器增加
    frame_count += 1

# 释放摄像头资源
cap.release()
# 关闭所有OpenCV窗口
cv2.destroyAllWindows()