python 眼球跟踪识别代码

以下是一个简单的Python眼球跟踪识别代码，需要使用OpenCV和dlib库：

import cv2
import dlib

# 初始化dlib的人脸检测器和眼睛检测器
detector = dlib.get_frontal_face_detector()
predictor = dlib.shape_predictor("shape_predictor_68_face_landmarks.dat")

# 打开电脑摄像头
cap = cv2.VideoCapture(0)

while True:
    # 读取视频帧
    ret, frame = cap.read()

    # 转换为灰度图像
    gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)

    # 人脸检测
    faces = detector(gray)

    for face in faces:
        # 眼睛检测
        landmarks = predictor(gray, face)
        left_eye = []
        right_eye = []

        for n in range(36, 42):
            x = landmarks.part(n).x
            y = landmarks.part(n).y
            left_eye.append((x, y))

        for n in range(42, 48):
            x = landmarks.part(n).x
            y = landmarks.part(n).y
            right_eye.append((x, y))

        # 计算眼睛中心
        left_eye_center = (int((left_eye[0][0] + left_eye[3][0]) / 2), int((left_eye[1][1] + left_eye[4][1]) / 2))
        right_eye_center = (int((right_eye[0][0] + right_eye[3][0]) / 2), int((right_eye[1][1] + right_eye[4][1]) / 2))

        # 绘制眼睛中心
        cv2.circle(frame, left_eye_center, 5, (0, 0, 255), -1)
        cv2.circle(frame, right_eye_center, 5, (0, 0, 255), -1)

    # 显示视频帧
    cv2.imshow('frame', frame)

    # 按下q键退出
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

# 释放资源
cap.release()
cv2.destroyAllWindows()

该代码通过使用dlib库进行人脸和眼睛检测，计算出眼睛中心并在视频帧上绘制出来，可以实现简单的眼球跟踪识别。