opencv基于camshift的人脸跟踪

基于 CamShift 算法的人脸跟踪是一种常见的计算机视觉应用，可以利用 OpenCV 库轻松实现。下面是一个简单的基于 CamShift 算法的人脸跟踪代码示例：

import cv2

# 加载人脸分类器
face_cascade = cv2.CascadeClassifier('haarcascade_frontalface_default.xml')

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

# 获取第一帧，并找到其中的人脸
ret, frame = cap.read()
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
faces = face_cascade.detectMultiScale(gray, 1.3, 5)

# 选取第一个人脸作为跟踪目标
if len(faces) > 0:
    (x, y, w, h) = faces[0]
    track_window = (x, y, w, h)
    roi = gray[y:y+h, x:x+w]
    hsv_roi = cv2.cvtColor(cv2.merge([roi, roi, roi]), cv2.COLOR_BGR2HSV)
    hist = cv2.calcHist([hsv_roi], [0], None, [180], [0, 180])
    cv2.normalize(hist, hist, 0, 255, cv2.NORM_MINMAX)
    term_crit = (cv2.TERM_CRITERIA_EPS | cv2.TERM_CRITERIA_COUNT, 10, 1)

while True:
    ret, frame = cap.read()
    if ret == True:
        # 将当前帧转换为 HSV 颜色空间
        hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
        
        # 计算反向投影
        dst = cv2.calcBackProject([hsv], [0], hist, [0, 180], 1)
        
        # 应用 CamShift 算法进行跟踪
        ret, track_window = cv2.CamShift(dst, track_window, term_crit)
        
        # 画出跟踪框
        pts = cv2.boxPoints(ret)
        pts = np.int0(pts)
        img = cv2.polylines(frame, [pts], True, 255, 2)
        
        # 显示结果
        cv2.imshow('img', img)
        if cv2.waitKey(1) & 0xFF == ord('q'):
            break

# 释放摄像头并关闭窗口
cap.release()
cv2.destroyAllWindows()

在上述代码中，首先加载了人脸分类器，然后打开了摄像头获取第一帧图像。接着使用人脸分类器检测第一帧图像中的人脸，并选取第一个人脸作为跟踪目标。计算目标的颜色直方图，并设置 CamShift 算法的终止准则。随后进入循环，读取每一帧图像，并将其转换为 HSV 颜色空间。计算当前帧图像的反向投影，并使用 CamShift 算法进行跟踪。最后将跟踪结果画出，并显示在窗口中。