一些小细节代码笔记汇总

咚咚锵咚咚锵2024-11-03 2:10

Python

cv2抓取摄像头图片保存到本地

python 复制代码

import cv2
import datetime, os

savePath = "E:/Image/"

if not os.path.exists(savePath):
    os.makedirs(savePath)

cap = cv2.VideoCapture(0)
capture = False

if not cap.isOpened():
    print("无法打开摄像头")
    exit()

while True:
    ret, frame = cap.read()
    
    if not ret:
        print("无法接收帧 (stream end?). Exiting ...")
        break

    cv2.imshow('Image', frame)
    current_time = datetime.datetime.now().strftime("%Y%m%d_%H%M%S_%f")
    if capture:
        cv2.imwrite(os.path.join(savePath, current_time + ".jpg"), frame)
    
    key = cv2.waitKey(1) & 0xFF
    # 按下'q'键退出循环
    if key == ord('q'):
        break
    elif key == ord('s'):
        print("开始抓取图片")
        capture = True

cap.release()
cv2.destroyAllWindows()

cv2.waitKey(1) & 0xFF的作用是：

cv2.waitKey(1)：等待1毫秒，检查是否有键盘输入。如果有输入，则返回按键的ASCII码值；如果没有输入，则返回-1。
& 0xFF：通过按位与操作，将返回值的高位清零，只保留低8位。这是因为在某些系统上，cv2.waitKey可能返回一个大于8位的整数，而我们只关心低8位的ASCII码值。

这样做的目的是确保代码在不同平台上都能正确处理键盘输入。例如，在Windows和Linux系统上，cv2.waitKey的返回值可能有所不同，通过& 0xFF可以统一处理这些差异。