/lstm_pm_torch

lstm pose machine pytorch implementation

Primary LanguagePython

Implementation of LSTM Pose Machines with Pytorch(revised version)

This is pytorch implement of LSTM Pose Machines

Note: Uses a custom dataset as per the original uploader.

Prerequisites

  • Python 2.7
  • scipy
  • sklearn
  • pillow
  • PyTorch 0.3
  • torchvision 0.1.9
  • pandas
  • numpy

Train

python lstm_pm_train.py

Test

python test_lstm_pm.py

References

LSTM Pose Machines

lawy623/LSTM_Pose_Machines