這是一個對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踐行者 的文章