• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

Project: FastMaskRCNN (GitHub Link)

• FastMaskRCNN-master
  • download_and_convert_data.py
  • libs
    • datasets
      • coco.py
      • download_and_convert_coco.py
      • pycocotools
        • coco.py
        • Makefile
        • common
          • maskApi.h
          • maskApi.c
          • gason.cpp
          • gason.h
        • setup.py
        • mask.py
        • cocoeval.py
        • __init__.py
        • _mask.pyx
      • __init__.py
      • dataset_factory.py
    • Makefile
    • layers
      • anchor.py
      • sample.py
      • wrapper.py
      • mask.py
      • crop.py
      • roi.py
      • __init__.py
      • assign.py
    • nms
      • gpu_nms.pyx
      • gpu_nms.hpp
      • cpu_nms.pyx
      • nms_kernel.cu
      • __init__.py
      • py_cpu_nms.py
      • .gitignore
    • memory_util.py
    • setup.py
    • boxes
      • cython_bbox_transform.pyx
      • nms.pyx
      • blob.py
      • gprof2dot.py
      • cython_bbox_transform.py
      • anchor.py
      • roi.py
      • __init__.py
      • profile
      • nms.py
      • nms_wrapper.py
      • cython_anchor.pyx
      • cython_bbox.py
      • .gitignore
      • timer.py
      • cython_nms.py
      • bbox.pyx
      • cython_anchor.py
      • bbox_transform.py
    • preprocessings
      • __init__.py
      • utils.py
      • coco_v1.py
    • nets
      • pyramid_network.py
      • train_utils.py
      • resnet_v1.py
      • nets_factory.py
      • resnet_utils.py
      • __init__.py
    • configs
      • config_v1.py
      • __init__.py
    • __init__.py
    • visualization
      • pil_utils.py
      • summary_utils.py
      • __init__.py
    • logs
      • log.py
      • __init__.py
    • make.sh
  • LICENSE
  • data
    • README.md
  • README.md
  • train
    • train_utils.py
    • train.py
    • __init__.py
  • .gitignore
  • unit_test
    • layer_test.py
    • data_test.py
    • __init__.py
    • resnet50_test.py
    • preprocessing_test.py

Mask RCNN

Mask RCNN in TensorFlow

This repo attempts to reproduce this amazing work by Kaiming He et al. : Mask R-CNN

Requirements

• Tensorflow (>= 1.0.0)
• Numpy
• COCO dataset
• Resnet50

How-to

1. Go to ./libs/datasets/pycocotools and run make
2. Download COCO dataset, place it into ./data, then run python download_and_convert_data.py to build tf-records. It takes a while.
3. 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/
4. Go to ./libs and run make
5. run python train/train.py for training
6. There are certainly some bugs, please report them back, and let's solve them together.

TODO:

• [x] ROIAlign
• [x] COCO Data Provider
• [x] Resnet50
• [x] Feature Pyramid Network
• [x] Anchor and ROI layer
• [x] Mask layer
• [x] Speedup anchor layer with cython
• [x] Combining all modules together.
• [x] Testing and debugging (in progress)
• [ ] Training / evaluation on COCO
• [ ] Add image summary to show some results
• [ ] Converting ResneXt
• [ ] Training >2 images

Call for contributions

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

Acknowledgment

This repo borrows tons of code from

• TFFRCNN
• py-faster-rcnn
• faster_rcnn
• tf-models

License

See LICENSE for details.