Mask RCNN in TensorFlow
This repo attempts to reporduce this amazing work by Kaiming He. Mask RCNN. The original work involves two stages, a pyramid Faster-RCNN for object detection and another network (with the same structure) for instance level segmentation. However, we aim to build an end-to-end framework in the first place.
- 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
- Anything helps this repo, including discussion, testing, promotion and of course your awesome code.