matplotlib绘制3D人体姿态

最新推荐文章于 2024-05-20 14:56:28 发布

不知名的好人

最新推荐文章于 2024-05-20 14:56:28 发布

阅读量239

点赞数

分类专栏： python 文章标签： matplotlib 3d python

本文链接：https://blog.csdn.net/ljh0408abc/article/details/132547008

版权

python 专栏收录该内容

2 篇文章 0 订阅

订阅专栏

效果图如下：

导入必要的库：

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

假设你已经读取了人体关键点的信息，此处直接用静态数据。

post_out = np.array([
    [0,0,0.787822068],
    [0.123238757,-0.040780518,0.78164047],
    [0.131162703,0.05052188,0.356339693],
    [0.101998687,-0.134696901,0],
    [-0.123237714,0.040785622,0.79400897],
    [-0.040740184,0.075435936,0.358784974],
    [-0.032220125,-0.093336225,0.009030402],
    [-0.004283257,-0.027451605,1.036530614],
    [-0.020511895,-0.014475852,1.293020725],
    [-0.035262831,0.071317613,1.387965202],
    [-0.07418853,0.023139238,1.495238662],
    [-0.176504895,0.024071693,1.260654569],
    [-0.167135298,0.210691005,1.104242444],
    [0.009177908,0.251894534,1.199225664],
    [0.130504638,-0.098547608,1.273252487],
    [0.233572304,-0.229689911,1.055113673],
    [0.307354271,-0.038980789,0.936925113],
])

创建图表对象

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

创建显示函数：

def show3Dpose(vals, ax):
    ax.view_init(elev=15., azim=70)

    lcolor=(0,0,1)
    rcolor=(1,0,0)

    I = np.array( [0, 0, 1, 4, 2, 5, 0, 7,  8,  8, 14, 15, 11, 12, 8,  9])
    J = np.array( [1, 4, 2, 5, 3, 6, 7, 8, 14, 11, 15, 16, 12, 13, 9, 10])

    LR = np.array([0, 1, 0, 1, 0, 1, 0, 0, 0,   1,  0,  0,  1,  1, 0, 0], dtype=bool)

    for i in np.arange( len(I) ):
        x, y, z = [np.array( [vals[I[i], j], vals[J[i], j]] ) for j in range(3)]
        ax.plot(x, y, z, lw=2, color = lcolor if LR[i] else rcolor)

    RADIUS = 0.72
    RADIUS_Z = 0.7

    xroot, yroot, zroot = vals[0,0], vals[0,1], vals[0,2]
    ax.set_xlim3d([-RADIUS+xroot, RADIUS+xroot])
    ax.set_ylim3d([-RADIUS+yroot, RADIUS+yroot])
    ax.set_zlim3d([-RADIUS_Z+zroot, RADIUS_Z+zroot])
    ax.set_aspect('auto') # works fine in matplotlib==2.2.2

    white = (1.0, 1.0, 1.0, 0.0)
    ax.xaxis.set_pane_color(white) 
    ax.yaxis.set_pane_color(white)
    ax.zaxis.set_pane_color(white)

    ax.tick_params('x', labelbottom = False)
    ax.tick_params('y', labelleft = False)
    ax.tick_params('z', labelleft = False)

调用函数，输出图像：

show3Dpose(post_out,ax)
plt.show()

完整代码

import numpy as np
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

post_out = np.array([
    [0,0,0.787822068],
    [0.123238757,-0.040780518,0.78164047],
    [0.131162703,0.05052188,0.356339693],
    [0.101998687,-0.134696901,0],
    [-0.123237714,0.040785622,0.79400897],
    [-0.040740184,0.075435936,0.358784974],
    [-0.032220125,-0.093336225,0.009030402],
    [-0.004283257,-0.027451605,1.036530614],
    [-0.020511895,-0.014475852,1.293020725],
    [-0.035262831,0.071317613,1.387965202],
    [-0.07418853,0.023139238,1.495238662],
    [-0.176504895,0.024071693,1.260654569],
    [-0.167135298,0.210691005,1.104242444],
    [0.009177908,0.251894534,1.199225664],
    [0.130504638,-0.098547608,1.273252487],
    [0.233572304,-0.229689911,1.055113673],
    [0.307354271,-0.038980789,0.936925113],
])

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')

def show3Dpose(vals, ax):
    ax.view_init(elev=15., azim=70)

    lcolor=(0,0,1)
    rcolor=(1,0,0)

    I = np.array( [0, 0, 1, 4, 2, 5, 0, 7,  8,  8, 14, 15, 11, 12, 8,  9])
    J = np.array( [1, 4, 2, 5, 3, 6, 7, 8, 14, 11, 15, 16, 12, 13, 9, 10])

    LR = np.array([0, 1, 0, 1, 0, 1, 0, 0, 0,   1,  0,  0,  1,  1, 0, 0], dtype=bool)

    for i in np.arange( len(I) ):
        x, y, z = [np.array( [vals[I[i], j], vals[J[i], j]] ) for j in range(3)]
        ax.plot(x, y, z, lw=2, color = lcolor if LR[i] else rcolor)

    RADIUS = 0.72
    RADIUS_Z = 0.7

    xroot, yroot, zroot = vals[0,0], vals[0,1], vals[0,2]
    ax.set_xlim3d([-RADIUS+xroot, RADIUS+xroot])
    ax.set_ylim3d([-RADIUS+yroot, RADIUS+yroot])
    ax.set_zlim3d([-RADIUS_Z+zroot, RADIUS_Z+zroot])
    ax.set_aspect('auto') # works fine in matplotlib==2.2.2

    white = (1.0, 1.0, 1.0, 0.0)
    ax.xaxis.set_pane_color(white) 
    ax.yaxis.set_pane_color(white)
    ax.zaxis.set_pane_color(white)

    ax.tick_params('x', labelbottom = False)
    ax.tick_params('y', labelleft = False)
    ax.tick_params('z', labelleft = False)

show3Dpose(post_out,ax)
plt.show()