Realtime Human Pose Estimation
Project description
PoseCamera
PoseCamera is python based SDK for multi human pose estimation through RGB webcam.
Install
install posecamera package through pip
pip install posecamera
If you are having issues with the installation on Windows OS then check this page
Usage
See Google colab notebook https://colab.research.google.com/drive/18uoYeKmliOFV8dTdOrXocClCA7nTwRcX?usp=sharing
draw pose keypoints on image
import posecamera
import cv2
det = posecamera.pose_tracker.PoseTracker()
image = cv2.imread("example.jpg")
pose = det(image)
for name, (y, x, score) in pose.keypoints.items():
cv2.circle(image, (int(x), int(y)), 4, (255, 0, 0), -1)
cv2.imshow("PoseCamera", image)
cv2.waitKey(0)
output of the above example
or get keypoints array
for pose in poses:
keypoints = pose.keypoints
Handtracker
import posecamera
import cv2
det = posecamera.hand_tracker.HandTracker()
image = cv2.imread("tmp/hands.jpg")
keypoints, bbox = det(image)
for hand_keypoints in keypoints:
for (x, y) in hand_keypoints:
cv2.circle(image, (int(x), int(y)), 3, (255, 0, 0), -1)
cv2.imshow("PoseCamera - Hand Tracking", image)
cv2.waitKey(0)
Using Docker
The official docker image is hosted on Docker Hub. The very first step is to install the docker docker on your system.
Also note that your Nvidia driver Needs to be compatible with CUDA10.2.
Doing inference on live webcam feed.
xhost +; docker run --name posecamera --rm --net=host --gpus all -e DISPLAY=$DISPLAY --device=/dev/video0:/dev/video0 wondertree/posecamera --video=0
GPU & Webcam support (if running docker) is not available on Windows Operating System.
To run inference on images use the following command.
docker run --name posecamera --rm --net=host -e DISPLAY=$DISPLAY wondertree/posecamera --images ./tmp/female_pose.jpg --cpu
For more details read our Docs
The base of this repository is based on the following research paper.
@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}
}
Share your thoughts
Join our Discussion Channel! We love to hear your ideas, suggestions or pull request
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for posecamera-2.1.2-py2.py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | ebae1ebc5ff2cb7ff7b368fcabc78491266057f0d66190757ae2eee1504f3f45 |
|
MD5 | 8cce1f7a2bec178c7e65bc1ccf3466e1 |
|
BLAKE2b-256 | 883f71c2921bc98e6b63fb238a18816bda92144b75d8a6793f09a62f63ca417f |