加入CODE CHINA

· 不限速    · 不限空间    · 不限人数    · 私仓免费

免费加入
README.md

Mask RCNN

Mask RCNN in TensorFlow

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

Requirements

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:

  • 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
  • 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

License

See LICENSE for details.

项目简介

🚀 Github 镜像仓库 🚀

源项目地址

https://github.com/charlesshang/fastmaskrcnn

发行版本

当前项目没有发行版本

贡献者 10

开发语言

  • Python 93.5 %
  • C++ 3.0 %
  • C 2.4 %
  • Cuda 1.1 %
  • Makefile 0.1 %