这是一个对DeepPose的复现。实现的论文是Human Pose Estimation via Deep Neural Networks.
代码位置github.com/mitmul/deeppose。
依赖
Python 3.5.1+
◦Chainer 1.13.0+
◦numpy 1.9+
◦scikit-image 0.11.3+
◦OpenCV 3.1.0+
安装:
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
数据准备:
bash datasets/download.sh
python datasets/flic_dataset.py
python datasets/lsp_dataset.py
python datasets/mpii_dataset.py
训练:
对FLIC数据集
bash shells/train_flic.sh
对MPII数据集
bash shells/train_flic.sh
閱讀更多 AI踐行者 的文章