人体关键点图例
参考:https://google.github.io/mediapipe/solutions/pose.html
函数式
import cv2
import mediapipe as mp
import time
import os
import random
def video_ope(file):
frame = 0
switch = True
cap = cv2.VideoCapture(file)
mypose= mp.solutions.pose
pose = mypose.Pose()
myDraw = mp.solutions.drawing_utils
DrawingSpec_point = myDraw.DrawingSpec((0, 255, 0), 2, 2)
while(True):
ret, img = cap.read()
img_rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
results = pose.process(img_rgb)
#print(results.pose_landmarks)
if (results.pose_landmarks):
for id,lm in enumerate(results.pose_landmarks.landmark):
#print (id,"\n",lm)
h,w,c = img.shape
cx,cy = int(lm.x*w),int(lm.y*h)
print (id,cx,cy)
if id == 0 :
cv2.circle(img,(cx,cy),12,(0,255,255),cv2.FILLED)
else:
cv2.circle(img,(cx,cy),7,(255,0,0),cv2.FILLED)
myDraw.draw_landmarks(img,results.pose_landmarks,mypose.POSE_CONNECTIONS,DrawingSpec_point)
frame +=1
cv2.putText(img,str(int(frame)),(10,70),cv2.FONT_HERSHEY_PLAIN,3,(255,0,255),3)
cv2.imshow("Frame", img)
key = cv2.waitKey(1) & 0xFF
if key== ord('q'):
cv2.waitKey(0)
if key== 27:
break
cap.release()
cv2.destroyAllWindows()
def main():
file = "video/anime.mp4"
video_ope(file)
if __name__=="__main__":
main()
模块式
import cv2
import mediapipe as mp
import time
import os
import random
class PoseDetector:
def __init__(self, mode = False, upBody = False, smooth=True, detectionCon = 0.5, trackCon = 0.5):
self.mode = mode
self.upBody = upBody
self.smooth = smooth
self.detectionCon = detectionCon
self.trackCon = trackCon
self.mpDraw = mp.solutions.drawing_utils
self.mpPose = mp.solutions.pose
self.DrawingSpec_point = self.mpDraw.DrawingSpec((0, 255, 0), 2, 2)
if mode:
self.pose = self.mpPose.Pose(self.mode, self.upBody, self.smooth, self.detectionCon, self.trackCon)
else:
self.pose = self.mpPose.Pose()
def findPose(self, img, draw=True):
imgRGB = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
self.results = self.pose.process(imgRGB)
#print(results.pose_landmarks)
if self.results.pose_landmarks:
if draw:
self.mpDraw.draw_landmarks(img, self.results.pose_landmarks, self.mpPose.POSE_CONNECTIONS,self.DrawingSpec_point)
return img
def getPosition(self, img, draw=True):
lmList= []
if self.results.pose_landmarks:
for id, lm in enumerate(self.results.pose_landmarks.landmark):
h, w, c = img.shape
cx, cy = int(lm.x * w), int(lm.y * h)
lmList.append([id, cx, cy])
if draw:
if id == 0 :
cv2.circle(img,(cx,cy),12,(255,255,20),cv2.FILLED)
else:
cv2.circle(img,(cx,cy),5,(255,0,0),cv2.FILLED)
if draw:
cv2.circle(img, (cx, cy), 5, (255, 0, 0), cv2.FILLED)
return lmList
def main(file):
cap = cv2.VideoCapture(file)
frame = 0
detector = PoseDetector()
while True:
success, img = cap.read()
img = detector.findPose(img)
lmList = detector.getPosition(img)
frame+=1
cv2.putText(img, str(int(frame)), (50, 50), cv2.FONT_HERSHEY_SIMPLEX, 1, (255, 0, 0), 3)
cv2.imshow("Image", img)
#cv2.waitKey(1)
key = cv2.waitKey(1) & 0xFF
if key== ord('q'):
cv2.waitKey(0)
if key== 27:
break
cap.release()
cv2.destroyAllWindows()
if __name__=="__main__":
file = "video/anime.mp4"
main(file)