lstm_pm_pytorch
lstm_pm_pytorch copied to clipboard
implementation of LSTM Pose Machines with Pytorch
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