Skip to content

FangYang970206/MaskRCNN-Keypoint-Demo

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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 Colab

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

demo.gif

Releases

No releases published

Packages

No packages published