This repo is about Mask RCNN (paper: https://arxiv.org/abs/1703.06870) with human-keypoint for human pose estimation. This repo relies on FangYang970206's repo.
- Python 3.4+
- TensorFlow 1.3+
- keras 2.0.8+
- numpy, skimage, scipy, Pillow, cython, h5py
- cocoapi:
pip install "git+https://github.com/philferriere/cocoapi.git#egg=pycocotools&subdirectory=PythonAPI"
- cv2:
pip install opencv-python
You can open the notebook in google colab, the environment can prepare in one cell and has free GPU.Just enjoy!
first, clone this repo,
$ git clone https://github.com/Krupal09/MaskRCNN-Demo
then,
$ cd MaskRCNN-Keypoint-Demo
download the pre-trained model(dropbox) in the MaskRCNN-Keypoint-Demo folder.
finally :
$ python main.py --image path/to/image