DeepPose implementation in Chainer
Switch branches/tags
Nothing to show
Clone or download
mitmul Merge pull request #35 from unnonouno/fix-matrix
Use list instead of str for initializer of matrix object
Latest commit fcf53af Jan 13, 2017
Permalink
Failed to load latest commit information.
datasets Modify README Aug 22, 2016
models Modify README Aug 22, 2016
scripts Use list instead of str for initializer of matrix object Jan 11, 2017
shells Modify README Aug 22, 2016
tests Modify README Aug 22, 2016
.gitignore Fix and refactored some files Jun 30, 2016
LICENSE Initial commit Nov 11, 2014
README.md Modify README Aug 22, 2016

README.md

DeepPose

NOTE: This is not official implementation. Original paper is DeepPose: Human Pose Estimation via Deep Neural Networks.

Requirements

  • Python 3.5.1+

I strongly recommend to use Anaconda environment. This repo may be able to be used in Python 2.7 environment, but I haven't tested.

Installation of dependencies

pip install chainer
pip install numpy
pip install scikit-image
# for python3
conda install -c https://conda.binstar.org/menpo opencv3
# for python2
conda install opencv

Dataset preparation

bash datasets/download.sh
python datasets/flic_dataset.py
python datasets/lsp_dataset.py
python datasets/mpii_dataset.py

MPII Dataset

  • MPII Human Pose Dataset
  • training images: 18079, test images: 6908
    • test images don't have any annotations
    • so we split trining imges into training/test joint set
    • each joint set has
  • training joint set: 17928, test joint set: 1991

Start training

Starting with the prepared shells is the easiest way. If you want to run train.py with your own settings, please check the options first by python scripts/train.py --help and modify one of the following shells to customize training settings.

For FLIC Dataset

bash shells/train_flic.sh

For LSP Dataset

bash shells/train_lsp.sh

For MPII Dataset

bash shells/train_mpii.sh

GPU memory requirement

  • AlexNet
    • batchsize: 128 -> about 2870 MiB
    • batchsize: 64 -> about 1890 MiB
    • batchsize: 32 (default) -> 1374 MiB
  • ResNet50
    • batchsize: 32 -> 6877 MiB

Prediction

Will add some tools soon