1292765944 / FastMaskRCNN

Mask RCNN in TensorFlow

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Fast Mask RCNN

Mask RCNN in TensorFlow

See you in one week

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.

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 (I feel a little bit lucky..)
  • Training / evaluation on COCO
  • Other backbone networks
  • Training >2 images

Call for contribution

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

About

Mask RCNN in TensorFlow

License:Apache License 2.0


Languages

Language:Python 91.0%Language:C++ 4.1%Language:C 3.2%Language:Cuda 1.6%Language:Makefile 0.1%Language:Shell 0.0%