Skip to content

Wangsujeon/FastMaskRCNN

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Mask RCNN

Mask RCNN in TensorFlow

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

Requirements

How-to

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

TODO:

  • ROIAlign
  • COCO Data Provider
  • Resnet50
  • Feature Pyramid Network
  • Anchor and ROI layer
  • Mask layer
  • Speedup anchor layer with cython
  • Combining all modules together.
  • 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.

Acknowledgment

This repo borrows tons of code from

License

See LICENSE for details.

About

Mask RCNN in TensorFlow

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Python 93.1%
  • C++ 3.1%
  • C 2.5%
  • Cuda 1.2%
  • Other 0.1%