A PyTorch toolkit for 2D Human Pose Estimation
PyTorch-Pose
PyTorch-Pose is a PyTorch implementation of the general pipeline for 2D single human pose estimation. The aim is to provide the interface of the training/inference/evaluation, and the dataloader with various data augmentation options for the most popular human pose databases (e.g., the MPII human pose, LSP and FLIC).
Some codes for data preparation and augmentation are brought from the Stacked hourglass network. Thanks to the original author.
Update: this repository is compatible with PyTorch 0.4.1/1.0 now!
Features
- Multi-thread data loading
- Multi-GPU training
- Logger
- Training/testing results visualization
Installation
-
PyTorch (>= 0.4.1): Please follow the installation instruction of PyTorch. Note that the code is developed with Python2 and has not been tested with Python3 yet.
-
Clone the repository