/FastMaskRCNN

Mask RCNN in TensorFlow

Primary LanguagePythonApache License 2.0Apache-2.0

Fast Mask RCNN

Mask RCNN in TensorFlow
This repo attempts to reporduce this amazing work by Kaiming He. Mask RCNN.

Known Problems

  • Allocate-Zero-Memory Error, solved by @santisy
  • Allowing larger initial learning rate.
  • Chooseing proper initial weights.

How-to

  1. Download coco dataset, place it into ./data, then run python download_and_convert_data.py to build tf-record. It takes a while.
  2. 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/
  3. run python test/resnet50_test.py for training
  4. There are certainly some bugs, please report them back, and let's solve them togather.

Timeline

  • 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
  • Data queue
  • Data agument
  • Other backbone networks
  • Training >2 images

Call for contribution

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