/flowtrack.pytorch

Pytorch implementation of FlowTrack (Simple Baselines for Human Pose Estimation and Tracking).

Primary LanguagePython

flowtrack.pytorch

Pytorch implementation of FlowTrack.

Simple Baselines for Human Pose Estimation and Tracking (https://arxiv.org/pdf/1804.06208.pdf)

TO DO:

  • Human detection
  • Single person pose estimation
    • Building
    • Training
  • Optical flow estimation
  • Box propagation
  • Pose tracking

Installation

cd lib
./make.sh

Demo

Detection

Download pretrained detection model into models/detection/. Refer to pytorch-faster-rcnn for more information.

python ./tools/detection/demo.py

Optical Flow Estimation

Download pretrained flownet into models/flownet/. Refer to flownet2-pytorch for more information.

python ./tools/flownet/demo.py --model </path/to/model>