xilaili / FastMaskRCNN

Mask RCNN in TensorFlow

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.