adirik / lightweight-openpose

PyTorch version of Lightweight OpenPose as introduced in "Real-time 2D Multi-Person Pose Estimation on CPU: Lightweight OpenPose"

  • Public
  • 1.6K runs
  • GitHub
  • Paper
  • License

😵 Uh oh! This model can't be run on Replicate because it was built with a version of Cog or Python that is no longer supported. Consider opening an issue on the model's GitHub repository to see if it can be updated to use a recent version of Cog. If you need any help, please Contact us about it.

Run time and cost

This model costs approximately $0.00056 to run on Replicate, or 1785 runs per $1, but this varies depending on your inputs. It is also open source and you can run it on your own computer with Docker.

This model runs on Nvidia T4 GPU hardware. Predictions typically complete within 3 seconds.

Readme

Lightweight OpenPose

Demo for Real-time 2D Multi-Person Pose Estimation on CPU: Lightweight OpenPose as implemented in the original repository. The available weights are downloaded from the link provided at the Github repository of this project.

API Usage

The API returns the 18 human body keypoints per person in the input image. The output data is as follows:

  • keypoint_names: Ordered list of 18 keypoint names for the human body
  • json_data: A dictionary containing the id, confidence, bbox (x1, y1, width, height), keypoints (list of all keypoints - [x, y] pairs) of each detected person.
    • Invisible / undetectable keypoints are returned as [-1, -1].
  • keypoints_img (optional): Input image overlayed with detected skeletons.

References

@inproceedings{osokin2018lightweight_openpose,
    author={Osokin, Daniil},
    title={Real-time 2D Multi-Person Pose Estimation on CPU: Lightweight OpenPose},
    booktitle = {arXiv preprint arXiv:1811.12004},
    year = {2018}
}