TropComplique / MultiPoseNet

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

MultiPoseNet in tensorflow (work in progress 🔧)

This an implementation of MultiPoseNet: Fast Multi-Person Pose Estimation using Pose Residual Network.

How to use this

  1. Download COCO dataset.
  2. Run create_tfrecords.py.
  3. Run train_keypoints.py.
  4. Run train_person_detector.py.
  5. Run train_prn.py.
  6. Run create_pb.py.

Requirements

  1. tensorflow 1.15
  2. Pillow 6.1, opencv-python 4.1
  3. numpy 1.17, scipy 1.3
  4. matplotlib 3.1, tqdm 4.36
  5. pycocotools

About

License:MIT License


Languages

Language:Python 84.5%Language:Jupyter Notebook 15.5%