Awesome Open Source
Awesome Open Source


Mask RCNN in TensorFlow

This repo attempts to reproduce this amazing work by Kaiming He et al. : Mask R-CNN



  1. Go to ./libs/datasets/pycocotools and run make
  2. Download COCO dataset, place it into ./data, then run python to build tf-records. It takes a while.
  3. Download pretrained resnet50 model, wget, unzip it, place it into ./data/pretrained_models/
  4. Go to ./libs and run make
  5. run python train/ for training
  6. There are certainly some bugs, please report them back, and let's solve them together.


  • [x] ROIAlign
  • [x] COCO Data Provider
  • [x] Resnet50
  • [x] Feature Pyramid Network
  • [x] Anchor and ROI layer
  • [x] Mask layer
  • [x] Speedup anchor layer with cython
  • [x] Combining all modules together.
  • [x] Testing and debugging (in progress)
  • [ ] Training / evaluation on COCO
  • [ ] Add image summary to show some results
  • [ ] Converting ResneXt
  • [ ] Training >2 images

Call for contributions

  • Anything helps this repo, including discussion, testing, promotion and of course your awesome code.


This repo borrows tons of code from


See LICENSE for details.

Alternative Project Comparisons
Related Awesome Lists
Top Programming Languages
Top Projects

Get A Weekly Email With Trending Projects For These Topics
No Spam. Unsubscribe easily at any time.
Python (836,862
Tensorflow (22,592
Segmentation (8,327
Resnet (3,238
Anchor (2,377
Coco (2,364
Rcnn (1,345
Mask Rcnn (355
Faster Rcnn (301