A docker build file for CMU openpose with Python API support. Tested this on a AMD Laptop with NVIDIA GEFORCE GPU (Nov 2020).
This should work with most of the recent NVIDIA latest GPUs.
The original Dockerfile was bassed on: https://hub.docker.com/r/cwaffles/openpose
- Nvidia Docker runtime: https://github.com/NVIDIA/nvidia-docker#quickstart
- CUDA 10.0 or higher on your host, check with
nvidia-smi - CuDNN 7.5 or higher on your host.
- A web-camera connected to your host (this assumes /dev/video0 is exported)
# Clone the Dockerfile repository
git clone https://github.com/hmurari/openpose-docker.git
# Build Docker container locally
cd openpose-docker
docker built -t openpose .
# Run the Docker container (share display, ipc, webcam with container)
sudo xhost +si:localuser:root
docker run --gpus all -e DISPLAY=$DISPLAY -v /tmp/.X11-unix/:/tmp/.X11-unix --ipc=host --device=/dev/video0:/dev/video0 -it openpose:latest /bin/bash
# Run Openpose body detection from webcam video
python3 001_body_from_camera.py
# Pull Docker image
docker pull hmurari/openpose-docker:latest
# Run container
sudo xhost +si:localuser:root
docker run --gpus all -e DISPLAY=$DISPLAY -v /tmp/.X11-unix/:/tmp/.X11-unix --ipc=host --device=/dev/video0:/dev/video0 -it openpose:latest /bin/bash
# Run Openpose body detection from webcam video
python3 001_body_from_camera.py