Mask RCNN in TensorFlow
Mask RCNN in TensorFlow
This repo attempts to reproduce this amazing work by Kaiming He et al. : Mask R-CNN
./libs/datasets/pycocotools
and run make
./data
, then run python download_and_convert_data.py
to build tf-records. It takes a while.wget http://download.tensorflow.org/models/resnet_v1_50_2016_08_28.tar.gz
, unzip it, place it into ./data/pretrained_models/
./libs
and run make
python train/train.py
for trainingThis repo borrows tons of code from
See LICENSE for details.