December 31, 2019

516 words 3 mins read



Deep Pose Estimation implemented using Tensorflow with Custom Architectures for fast inference.

repo name ildoonet/tf-pose-estimation
repo link
language Python
size (curr.) 96387 kB
stars (curr.) 3028
created 2017-07-31
license Apache License 2.0


‘Openpose’, human pose estimation algorithm, have been implemented using Tensorflow. It also provides several variants that have some changes to the network structure for real-time processing on the CPU or low-power embedded devices.

You can even run this on your macbook with a descent FPS!

Original Repo(Caffe) :

CMU’s Original Model on Macbook Pro 15" Mobilenet-thin on Macbook Pro 15" Mobilenet-thinon Jetson TX2
cmu-model mb-model-macbook mb-model-tx2
~0.6 FPS ~4.2 FPS @ 368x368 ~10 FPS @ 368x368
2.8GHz Quad-core i7 2.8GHz Quad-core i7 Jetson TX2 Embedded Board

Implemented features are listed here : features

Important Updates



You need dependencies below.

Pre-Install Jetson case

$ sudo apt-get install libllvm-7-ocaml-dev libllvm7 llvm-7 llvm-7-dev llvm-7-doc llvm-7-examples llvm-7-runtime
$ export LLVM_CONFIG=/usr/bin/llvm-config-7 


Clone the repo and install 3rd-party libraries.

$ git clone
$ cd tf-pose-estimation
$ pip3 install -r requirements.txt

Build c++ library for post processing. See :

$ cd tf_pose/pafprocess
$ swig -python -c++ pafprocess.i && python3 build_ext --inplace

Package Install

Alternatively, you can install this repo as a shared package using pip.

$ git clone
$ cd tf-pose-estimation
$ python install  # Or, `pip install -e .`

Models & Performances


Download Tensorflow Graph File(pb file)

Before running demo, you should download graph files. You can deploy this graph on your mobile or other platforms.

  • cmu (trained in 656x368)
  • mobilenet_thin (trained in 432x368)
  • mobilenet_v2_large (trained in 432x368)
  • mobilenet_v2_small (trained in 432x368)

CMU’s model graphs are too large for git, so I uploaded them on an external cloud. You should download them if you want to use cmu’s original model. Download scripts are provided in the model folder.

$ cd models/graph/cmu
$ bash


Test Inference

You can test the inference feature with a single image.

$ python --model=mobilenet_thin --resize=432x368 --image=./images/p1.jpg

The image flag MUST be relative to the src folder with no “~”, i.e:

--image ../../Desktop

Then you will see the screen as below with pafmap, heatmap, result and etc.


Realtime Webcam

$ python --model=mobilenet_thin --resize=432x368 --camera=0

Apply TensoRT

$ python --model=mobilenet_thin --resize=432x368 --camera=0 --tensorrt=True

Then you will see the realtime webcam screen with estimated poses as below. This Realtime Result was recored on macbook pro 13" with 3.1Ghz Dual-Core CPU.

Python Usage

This pose estimator provides simple python classes that you can use in your applications.

See or as references.

e = TfPoseEstimator(get_graph_path(args.model), target_size=(w, h))
humans = e.inference(image)
image = TfPoseEstimator.draw_humans(image, humans, imgcopy=False)

If you installed it as a package,

import tf_pose
coco_style = tf_pose.infer(image_path)

ROS Support

See : etcs/


See : etcs/


See : etcs/

comments powered by Disqus