lstm_pm_pytorch

implementation of LSTM Pose Machines with Pytorch

This is pytorch implement of LSTM Pose Machines

Please note that this model is used for my own datasets. But the architecture and training are very close to the original model, please feel free to use it for your own project.

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