personlab-tf icon indicating copy to clipboard operation
personlab-tf copied to clipboard

implementation of PersonLab(https://arxiv.org/abs/1803.08225) using TF-slim

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.

  • ubuntu 16.04
  • python 3.6
  • numpy 1.14.5
  • tensorflow 1.10.0
  • pycocotools 2.0.0
  • scikit-image 0.14.0

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)