python 利用摄像头识别水平度

在Python中，我们可以使用OpenCV库来实现摄像头识别水平角度的功能。OpenCV是一个开源计算机视觉库，它包含了许多图像处理和机器视觉算法，包括特征检测、人脸识别等。要测量水平角度，通常会采用以下几个步骤：

1. **安装依赖**：首先需要安装`opencv-python`库，可以使用pip安装：`pip install opencv-python`

2. **视频流获取**：通过`cv2.VideoCapture`函数打开摄像头，并创建一个VideoCapture对象。

import cv2

cap = cv2.VideoCapture(0)  # 使用默认摄像头，编号可以改为其他

3. **读取帧并预处理**：循环读取每一帧，将其转换为灰度图，然后使用霍夫变换检测直线，其中水平线应该是一条接近垂直的线。

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

    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
    edges = cv2.Canny(gray, threshold1=50, threshold2=150)

    lines = cv2.HoughLinesP(edges, rho=1, theta=np.pi/180, threshold=100, minLineLength=50, maxLineGap=10)
    for line in lines:
        x1, y1, x2, y2 = line[0]
        slope = (y2 - y1) / (x2 - x1)  # 计算斜率，近似为垂直则斜率为无穷大
        if abs(slope) > 0.95:  # 如果斜率非常接近于1，认为是水平线
            angle = np.arctan(abs(slope)) * 180 / np.pi  # 转换为角度
            print(f"当前水平角度约为: {angle} 度")

4. **关闭摄像头**：完成计算后，记得关闭摄像头释放资源。

cap.release()
cv2.destroyAllWindows()