Sign inSign up

dgrnwd/openpose-gpu

By dgrnwd

Updated over 7 years ago

CMU open pose built on the caffe container

Image
0

1.1K

dgrnwd/openpose-gpu repository overview

Openpose Dockerfile - Nvidia GPU

This docker file is built on top of the caffe gpu image. We rebuild caffe, knowing we have the correct requirements, as for some reason openpose does not find all the includes. The image includes all the available models at this time. Made available on Docker.

The image can be built with:

sudo docker build -t openpose:gpu .

There is now an image on Docker hub.

docker pull dgrnwd/openpose-gpu

Once the image is built we connect a data directory and run the image.

docker run --runtime=nvidia  -v /media/data1:/data -it openpose:gpu bash

The openpose binary is in the path, so we can call it:

openpose -helpshort

The output from openpose -help is saved in help.txt

An example for tracking a face might be:

openpose \
-display=0 \
-video /data/small-test.mov \
-face=true \
-face_net_resolution="368x368" \
-net_resolution="-1x640" \
-write_json="/data/op_out/kp_368_640" \
-write_video="/data/op_out/face_368_net_640.avi" \
-camera_fps=59.94

Another example, from the image pulled from docker hub.

sudo docker run --runtime=nvidia  \
    -v ~/emo-data:/data \
    -it dgrnwd/openpose-gpu:latest  bash



openpose \
-display=0 \
-video /data/aj-2k-sample.mov \
-frame_rotate=270 \
-face=true \
-write_json="/data/op_out/kp-aj-2k-sample-default" \
-write_video="/data/op_out/kp-aj-2k-sample-default.avi" \
-camera_fps=59.94

Tag summary

Content type

Image

Digest

Size

2.5 GB

Last updated

over 7 years ago

docker pull dgrnwd/openpose-gpu