Skip to content

scnuhealthy/Tensorflow_PersonLab

Repository files navigation

PersonLab

An Tensorflow implementation of PersonLab for Multi-Person Pose Estimation and Instance Segmentation. Identify every person instance, localize its facial and body keypoints, and estimate its instance segmentation mask.

Introduction

Code repo for reproducing 2018 ECCV paper PersonLab: Person Pose Estimation and Instance Segmentation with a Bottom-Up, Part-Based, Geometric Embedding Model

Result

Pose

pose

Segmentation

segmentation

Require

  • Python3

  • Tensorflow 1.80

  • pycocotools 2.0

  • skimage 0.13.0

  • python-opencv 3.4.1

Demo

  • Download the model
  • python demo.py to run the demo and visualize the model result

Training

Technical Debts

The augmentation code (which is different from the procedure in the PersonLab paper) and data iterator code is heavily borrowed from this fork of the Keras implementation of CMU's "Realtime Multi-Person Pose Estimation". (The pose plotting function is also influenced by the one in that repo.)

The mask generation code and visualization code are from this fork of the Keras implementation of PersonLab.

Citation

@inproceedings{papandreou2018personlab,
  title={PersonLab: Person pose estimation and instance segmentation with a bottom-up, part-based, geometric embedding model},
  author={Papandreou, George and Zhu, Tyler and Chen, Liang-Chieh and Gidaris, Spyros and Tompson, Jonathan and Murphy, Kevin},
  booktitle={Proceedings of the European Conference on Computer Vision (ECCV)},
  pages={269--286},
  year={2018}
}

About

Tensorflow implementation of PersonLab (https://arxiv.org/abs/1803.08225)

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages