Mask RCNN in TensorFlow
This repo attempts to reproduce this amazing work by Kaiming He et al. :
- Go to
./libs/datasets/pycocotools and run
- Download COCO dataset, place it into
./data, then run
python download_and_convert_data.py to build tf-records. It takes a while.
- Download pretrained resnet50 model,
wget http://download.tensorflow.org/models/resnet_v1_50_2016_08_28.tar.gz, unzip it, place it into
- Go to
./libs and run
python train/train.py for training
- There are certainly some bugs, please report them back, and let's solve them together.
Call for contributions
- Anything helps this repo, including discussion, testing, promotion and of course your awesome code.
This repo borrows tons of code from
See LICENSE for details.