import cv2
import numpy as np
import mediapipe as mp
# mediapipe入门
# 画图必备
mp_drawing = mp.solutions.drawing_utils
# 默认绘图风格
mp_drawing_styles = mp.solutions.drawing_styles
# 自定义绘图风格 参数：1、颜色，2、线条粗细，3、点的半径
DrawingSpec_point = mp_drawing.DrawingSpec((0, 255, 0), 1, 1)
DrawingSpec_line = mp_drawing.DrawingSpec((0, 0, 255), 1, 1)
# 导入方法
mp_face_detection = mp.solutions.face_detection # 人脸检测
mp_face_mesh = mp.solutions.face_mesh # 面网格
mp_hand = mp.solutions.hands # 手
mp_pose = mp.solutions.pose # 姿势
# 参数设置，使用"with 等号右侧 as 等号左侧:“，简便且安全
mode_face_detection = mp_face_detection.FaceDetection(model_selection=1, min_detection_confidence=0.5)
mode_face_mesh = mp_face_mesh.FaceMesh(static_image_mode=True, max_num_faces=1, refine_landmarks=True,
min_detection_confidence=0.5)
mode_hand = mp_hand.Hands(static_image_mode=True, max_num_hands=2, min_detection_confidence=0.5)
mode_pose = mp_pose.Pose(min_detection_confidence=0.5, min_tracking_confidence=0.5)
'''
with mp_face_detection.FaceDetection(model_selection=1, min_detection_confidence=0.5) as mode_face_detection:
with mp_face_mesh.FaceMesh(static_image_mode=True, max_num_faces=1, refine_landmarks=True,
min_detection_confidence=0.5) as mode_face_mesh:
with mp_hand.Hands(static_image_mode=True, max_num_hands=2, min_detection_confidence=0.5) as mode_hand:
with mp_pose.Pose(static_image_mode=True, model_complexity=2, enable_segmentation=True,
min_detection_confidence=0.5) as mode_pose:
'''
# 开启摄像头
cap = cv2.VideoCapture(0)
# 判断开启摄像头是否成功
while cap.isOpened():
success, img = cap.read()
# 玩一玩，打开摄像头失败时弹出图片
blank = np.zeros((720, 1280, 3))
if not success:
cv2.putText(blank, 'Please Check Your Camera', (150, 250), cv2.FONT_HERSHEY_TRIPLEX, 1.0, (0, 255, 0), 2)
cv2.imshow('Text', blank)
cv2.waitKey(1)
continue
img.flags.writeable = False
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) # 增强对于视觉的识别能力
# RGB图像处理
result_face_detection = mode_face_detection.process(img)
result_face_mesh = mode_face_mesh.process(img)
result_hand = mode_hand.process(img)
result_pose = mode_pose.process(img)
# 绘制关键点与连线
img.flags.writeable = True # 官方文件都有这个，还不理解作用是啥
img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) # 恢复原图像色彩
# 绘制脸部识别
if result_face_detection.detections:
for detection in result_face_detection.detections:
mp_drawing.draw_detection(img, detection)
# 绘制脸部网孔
if result_face_mesh.multi_face_landmarks:
for face_landmarks in result_face_mesh.multi_face_landmarks:
mp_drawing.draw_landmarks(image=blank, landmark_list=face_landmarks,
connections=mp_face_mesh.FACEMESH_TESSELATION,
connection_drawing_spec=mp_drawing_styles
.get_default_face_mesh_tesselation_style())
mp_drawing.draw_landmarks(image=blank, landmark_list=face_landmarks,
connections=mp_face_mesh.FACEMESH_CONTOURS,
connection_drawing_spec=mp_drawing_styles
.get_default_face_mesh_contours_style())
mp_drawing.draw_landmarks(image=blank, landmark_list=face_landmarks,
connections=mp_face_mesh.FACEMESH_IRISES,
connection_drawing_spec=mp_drawing_styles
.get_default_face_mesh_iris_connections_style())
# 绘制手部节点
if result_hand.multi_hand_landmarks:
for hand_landmarks in result_hand.multi_hand_landmarks:
mp_drawing.draw_landmarks(blank, hand_landmarks, mp_hand.HAND_CONNECTIONS,
mp_drawing_styles.get_default_hand_landmarks_style(),
mp_drawing_styles.get_default_hand_connections_style())
# 绘制身体节点
mp_drawing.draw_landmarks(blank, result_pose.pose_landmarks, mp_pose.POSE_CONNECTIONS,
landmark_drawing_spec=mp_drawing_styles.get_default_pose_landmarks_style())
# 翻转镜头，设置关闭条件
cv2.imshow('MediaPipe Face Detection', cv2.flip(blank, 1))
if cv2.waitKey(1) & 0xFF == ord('q'):
break
cap.release()
cv2.destroyAllWindows()
