Identify particular elements (like wrist, shoulder, knee...)

[JivanRoquet]

This library is really great work.

I was wondering, as it's not clear in the documentation, if there is a way to "simply" get the coordinates for the estimated location of different elements of the reconstituted pose mesh.

For example, let's say I'd like to get the following result:

{
    'wrist_R': (12, 22),
    'elbow_R': (43, 13),
    'shoulder_R': (44, 55),
    'neck': (56, 53)
}

Following through the steps in pose_mobile_net.ipynb, it doesn't like there's an immediate and/or obvious way to get to that sort of result.

In my use case, I'm actually not that interested in the "connections" between the dots, but much more in knowing what part of the body each particular dot refers to. Would you have idea of how to achieve that?

Thanks,

[michalfaber]

Hi @JivanRoquet I added a new folder estimation_example. The code does exactly what you need. Check out the scripts: coordinates.py, connections.py.