FangYang970206 / MaskRCNN-Keypoint-Demo

Mask RCNN for the human pose estimation

MaskRCNN-Keypoint-Demo

This repo is about Mask RCNN with human-keypoint. More info can be seen in this repo.

Requirements

• 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

Run in notebook

You can open the notebook in google colab, the environment can prepare in one cell and has free GPU.Just enjoy!

Run in local

first, clone this repo,

$ git clone https://github.com/FangYang970206/MaskRCNN-Keypoint-Demo

then,

$ cd MaskRCNN-Keypoint-Demo

download the pre-trained model(baiduyun,dropbox) in the MaskRCNN-Keypoint-Demo folder.

finally, you have two choices:

$ python main.py --image path/to/image

or

$ python main.py --video path/to/video

Example:

$ python main.py --image media/ski.jpg
$ python main.py --video media/human.mp4

Result