import cv2
import mediapipe as mp
import time
#第一步：保存mediapipe中的手勢(shì)識(shí)別解決方案到mpHands,hands,mpDraw中
mpHands = mp.solutions.hands #以下三個(gè)是mediapipe 中的API調(diào)用模板了
hands = mpHands.Hands(min_detection_confidence=0.5, min_tracking_confidence=0.5)#最小檢測(cè)置信度，最小追蹤置信度
mpDraw = mp.solutions.drawing_utils                              #獲取mediapipe解決方案的繪畫(huà)工具包
#第二步：參數(shù)設(shè)定
handLmsStyle = mpDraw.DrawingSpec(color=(0, 0, 255), thickness=3)#繪制手部關(guān)鍵點(diǎn)的顏色與粗細(xì)
handConStyle = mpDraw.DrawingSpec(color=(0, 255, 0), thickness=5)#繪制手部線(xiàn)條的顏色與粗細(xì)
pTime = 0 #和下面的cTime一起用于計(jì)算視頻輸入流FPS
cTime = 0
cap = cv2.VideoCapture(0) #打開(kāi)編號(hào)為0的攝像頭，這個(gè)一般是自帶攝像頭
#第三步：循環(huán)讀取視頻流到img，img輸入hands.hands函數(shù)得到結(jié)果，繪制結(jié)果到img并輸出
while True:
    ret, img = cap.read()       #從cap中讀取圖片到img，并將讀取是否成功的結(jié)果保存在ret
    if ret:
        imgRGB = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)  #模型訓(xùn)練的時(shí)候是使用RGB訓(xùn)練，對(duì)于這個(gè)類(lèi)型識(shí)別精度和速度比較高
        result = hands.process(imgRGB)                 #將RGB圖片輸入手部模型將結(jié)果保存在result
        # print(result.multi_hand_landmarks)           #打印result.multi_hand_landmarks內(nèi)容，可以去掉試一下
        imgHeight = img.shape[0]       #獲取攝像機(jī)圖片的高
        imgWidth = img.shape[1]        #獲取攝像機(jī)圖片的寬
        if result.multi_hand_landmarks:#如果multi_hand_landmarks不為空進(jìn)入循環(huán)
            for handLms in result.multi_hand_landmarks:  #遍歷multi_hand_landmarks內(nèi)每一個(gè)hand_landmark（手部關(guān)鍵點(diǎn)），相對(duì)于遍歷圖片中每一個(gè)手
                mpDraw.draw_landmarks(img, handLms, mpHands.HAND_CONNECTIONS, handLmsStyle, handConStyle) #調(diào)用mediapipe內(nèi)繪畫(huà)工具包繪畫(huà)手部關(guān)鍵點(diǎn)
                for i, lm in enumerate(handLms.landmark):  #i保存第幾個(gè)手部關(guān)鍵點(diǎn)，lm保存該點(diǎn)在圖中的歸一化值
                    xPos = int(lm.x * imgWidth) #第i個(gè)關(guān)鍵點(diǎn)x
                    yPos = int(lm.y * imgHeight)#第i個(gè)關(guān)鍵點(diǎn)y
                    cv2.putText(img, str(i), (xPos-25, yPos+5), cv2.FONT_HERSHEY_SIMPLEX, 0.4, (0, 0, 255), 2)#繪制關(guān)鍵點(diǎn)在，關(guān)鍵點(diǎn)左邊靠下一點(diǎn)
                    if i == 8:#當(dāng)畫(huà)到第八個(gè)關(guān)鍵點(diǎn)時(shí)
                         cv2.circle(img, (xPos, yPos), 10, (166, 0, 0), cv2.FILLED)#畫(huà)一個(gè)圓
                         #print(i, xPos, yPos)   #打印繪制這個(gè)點(diǎn)時(shí)候的坐標(biāo)
        cTime = time.time()
        fps = 1/(cTime-pTime)
        pTime = cTime
        cv2.putText(img, f"FPS : {int(fps)}", (30, 50), cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 0, 0), 3) #繪制FSP到圖中
        cv2.imshow('img', img) #輸出圖片
    if cv2.waitKey(1) == ord('q'):#點(diǎn)擊視頻，輸入q退出
        break

import cv2
import time
import mediapipe as mp
mp_drawing = mp.solutions.drawing_utils
mp_holistic = mp.solutions.holistic
holistic = mp_holistic.Holistic(min_detection_confidence=0.5, min_tracking_confidence=0.5)
handLmsStyle = mp_drawing.DrawingSpec(color=(0, 0, 255), thickness=0)#繪制手部關(guān)鍵點(diǎn)的顏色與粗細(xì)
handConStyle = mp_drawing.DrawingSpec(color=(0, 255, 0), thickness=4)#繪制手部線(xiàn)條的顏色與粗細(xì)
cap = cv2.VideoCapture(0)
while True:
    ret,image=cap.read()
    if ret:
        image = cv2.flip(image, 1)
        image=cv2.cvtColor(image,cv2.COLOR_BGR2RGB)
        results = holistic.process(image)
        if results.pose_landmarks:
            mp_drawing.draw_landmarks(image, results.face_landmarks, mp_holistic.FACEMESH_CONTOURS,handLmsStyle,handConStyle)
            mp_drawing.draw_landmarks(image, results.left_hand_landmarks, mp_holistic.HAND_CONNECTIONS)
            mp_drawing.draw_landmarks(image, results.right_hand_landmarks, mp_holistic.HAND_CONNECTIONS)
            mp_drawing.draw_landmarks(image, results.pose_landmarks, mp_holistic.POSE_CONNECTIONS)
        cv2.imshow("img",image)
        if cv2.waitKey(1)==ord("q"):
            break
holistic.close()