Commit 87059de5 authored by Yuxin Wu's avatar Yuxin Wu

update docs

parent a19eb489
...@@ -47,8 +47,6 @@ Speed: ...@@ -47,8 +47,6 @@ Speed:
a slow convolution algorithm, or you spend more time on autotune. a slow convolution algorithm, or you spend more time on autotune.
This is a general problem of TensorFlow when running against variable-sized input. This is a general problem of TensorFlow when running against variable-sized input.
3. With a large roi batch size (e.g. >= 256), GPU utilitization should stay above 90%. 3. This implementation is about 14% slower than detectron,
4. This implementation is about 14% slower than detectron,
probably due to the lack of specialized ops (e.g. AffineChannel, ROIAlign) in TensorFlow. probably due to the lack of specialized ops (e.g. AffineChannel, ROIAlign) in TensorFlow.
It's certainly faster than other TF implementation. It's certainly faster than other TF implementation.
# Faster-RCNN / Mask-RCNN on COCO # Faster-RCNN / Mask-RCNN on COCO
This example provides a minimal (only 1.6k lines) but faithful implementation the This example provides a minimal (only 1.6k lines) but faithful implementation of
following papers in combination: the following papers:
+ [Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks](https://arxiv.org/abs/1506.01497) + [Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks](https://arxiv.org/abs/1506.01497)
+ [Feature Pyramid Networks for Object Detection](https://arxiv.org/abs/1612.03144) + [Feature Pyramid Networks for Object Detection](https://arxiv.org/abs/1612.03144)
...@@ -30,7 +30,7 @@ DIR/ ...@@ -30,7 +30,7 @@ DIR/
## Usage ## Usage
Change config in `config.py`: Change config in `config.py`:
1. Change `BASEDIR` to `/path/to/DIR` as described above. 1. Change `BASEDIR` to `/path/to/DIR` as described above.
2. Change `MODE_MASK` to switch Faster-RCNN or Mask-RCNN. 2. Change `MODE_MASK`/`MODE_FPN`, or other options you like. Recommended configurations are listed in the table below.
Train: Train:
``` ```
...@@ -67,7 +67,7 @@ MaskRCNN results contain both bbox and segm mAP. ...@@ -67,7 +67,7 @@ MaskRCNN results contain both bbox and segm mAP.
|R101-C4 |512 |(800, 1333)|280k |40.1/34.4 |70h on 8 P100s| |R101-C4 |512 |(800, 1333)|280k |40.1/34.4 |70h on 8 P100s|
|R101-C4 |512 |(800, 1333)|360k |40.8/35.1 |63h on 8 V100s| |R101-C4 |512 |(800, 1333)|360k |40.8/35.1 |63h on 8 V100s|
The two R-50 360k models have the same configuration __and mAP__ The two R50-C4 360k models have the same configuration __and mAP__
as the `R50-C4-2x` entries in as the `R50-C4-2x` entries in
[Detectron Model Zoo](https://github.com/facebookresearch/Detectron/blob/master/MODEL_ZOO.md#end-to-end-faster--mask-r-cnn-baselines). [Detectron Model Zoo](https://github.com/facebookresearch/Detectron/blob/master/MODEL_ZOO.md#end-to-end-faster--mask-r-cnn-baselines).
So far this is the only TensorFlow implementation that can reproduce mAP in Detectron. So far this is the only TensorFlow implementation that can reproduce mAP in Detectron.
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment