python 计算图片中脸朝向角度

您可以使用Python中的dlib库，通过人脸关键点检测来计算图像中脸朝向的角度。以下是一个简单的示例代码：

import dlib
import cv2
import imutils

# 加载预训练的面部关键点检测器
face_detector = dlib.get_frontal_face_detector()
face_pose_predictor = dlib.shape_predictor("shape_predictor_68_face_landmarks.dat")

# 加载图像并缩放
image = cv2.imread("test.jpg")
image = imutils.resize(image, width=800)

# 在图像中检测人脸
faces = face_detector(image, 1)

# 对于每张检测到的人脸，计算脸部朝向角度
for face in faces:
    # 获取面部关键点并将其转换为 numpy array
    shape = face_pose_predictor(image, face)
    shape = dlib.full_object_detection.part(self=shape).shape_to_np()

    # 计算面部朝向角度
    dY = shape[0][1] - shape[16][1]
    dX = shape[0][0] - shape[16][0]
    angle = -np.degrees(np.arctan2(dY, dX))

    # 在图像中绘制角度和人脸框
    cv2.putText(image, "{:.2f}".format(angle), (10, 30), cv2.FONT_HERSHEY_SIMPLEX,
                0.7, (0, 0, 255), 2)
    cv2.rectangle(image, (face.left(), face.top()), (face.right(), face.bottom()),
                  (0, 0, 255), 2)

# 显示结果
cv2.imshow("Output", image)
cv2.waitKey(0)

这个代码可以读入一张名为"test.jpg"的图像，检测它中间的所有面孔，对于每个人脸计算脸部朝向角度，并在原始图像中绘制该角度和人脸框。注意：要运行此示例代码，您需要为dlib下载"shape_predictor_68_face_landmarks.dat"预训练模型。