opencv 人体姿态识别

OpenCV是一个流行的计算机视觉库，可以用于许多应用程序，包括人体姿态识别。在OpenCV中，可以使用OpenPose库来实现人体姿态估计。以下是使用Python和OpenCV实现人体姿态估计的步骤：

1.安装OpenCV和OpenPose库。

2.导入必要的库和模块。

3.加载图像并将其转换为灰度图像。

4.使用OpenPose库检测人体关键点。

5.将检测到的关键点连接起来以估计人体姿态。

6.显示结果图像。

下面是一个示例代码，演示如何使用Python和OpenCV实现人体姿态估计：

import cv2
import numpy as np

# 加载图像并将其转换为灰度图像
img = cv2.imread('test.jpg')
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

# 加载OpenPose模型
net = cv2.dnn.readNetFromTensorflow('graph_opt.pb')

# 设置参数
inWidth = 368
inHeight = 368
thr = 0.2

# 运行前向传递以检测人体关键点
inpBlob = cv2.dnn.blobFromImage(img, 1.0 / 255, (inWidth, inHeight), (0, 0, 0), swapRB=False, crop=False)
net.setInput(inpBlob)
output = net.forward()

# 获取关键点
H = output.shape[2]
W = output.shape[3]
points = []
for i in range(18):
    # 获取关键点的置信度图
    probMap = output[0, i, :, :]
    # 找到最大值的位置
    minVal, prob, minLoc, point = cv2.minMaxLoc(probMap)
    # 将关键点的位置缩放到原始图像大小
    x = (img.shape[1] * point[0]) / W
    y = (img.shape[0] * point[1]) / H
    # 如果置信度大于阈值，则将关键点添加到列表中
    if prob > thr:
        points.append((int(x), int(y)))
    else:
        points.append(None)

# 将关键点连接起来以估计人体姿态
pairs = [(0, 1), (1, 2), (2, 3), (3, 4), (1, 5), (5, 6), (6, 7), (1, 8), (8, 9), (9, 10), (10, 11), (8, 12), (12, 13), (13, 14), (0, 15), (15, 17), (0, 16), (16, 18), (14, 19), (19, 20), (14, 21), (11, 22), (22, 23), (11, 24)]
for pair in pairs:
    partA = pair[0]
    partB = pair[1]
    if points[partA] and points[partB]:
        cv2.line(img, points[partA], points[partB], (0, 255, 255), 2)

# 显示结果图像
cv2.imshow('Output', img)
cv2.waitKey(0)
cv2.destroyAllWindows()