sydsim / personlab-tf

implementation of PersonLab(https://arxiv.org/abs/1803.08225) using TF-slim
103 stars 19 forks source link
deep-learning human-pose-estimation python tensorflow

personlab-tf

This is an implemention of PersonLab using Tensorflow Slim. Resnet and Mobilenet in slim can be used as a pre-trained model. It is explained in the example notebook to understand how to use it.

Environment

this project has tested on this environment.

Pretrained Model

MobileNetV2 Based (Didn't measured the accuracy)

--- CAUTION ---

Currently, there is an issue that only appears on CPU environment. (https://github.com/sydsim/personlab-tf/issues/7)