Commit 9bfbb0a4 authored by Yuxin Wu's avatar Yuxin Wu

update maskrcnn notes

parent dc13638b
...@@ -25,8 +25,8 @@ DIR/ ...@@ -25,8 +25,8 @@ DIR/
## Usage ## Usage
Change config in `config.py`: Change config in `config.py`:
1. Set `BASEDIR` to `/path/to/DIR` as described above. 1. Change `BASEDIR` to `/path/to/DIR` as described above.
2. Set `MODE_MASK` to switch Faster-RCNN or Mask-RCNN. 2. Change `MODE_MASK` to switch Faster-RCNN or Mask-RCNN.
Train: Train:
``` ```
...@@ -48,22 +48,22 @@ Evaluate the performance of a model and save to json. ...@@ -48,22 +48,22 @@ Evaluate the performance of a model and save to json.
## Results ## Results
Models are trained on trainval35k and evaluated on minival using mAP@IoU=0.50:0.95. These models are trained with different configurations on trainval35k and evaluated on minival using mAP@IoU=0.50:0.95.
MaskRCNN results contain both bbox and segm mAP. MaskRCNN results contain both bbox and segm mAP.
|Backbone|`FASTRCNN_BATCH`|resolution |schedule|mAP (bbox/segm)|Time | |Backbone|`FASTRCNN_BATCH`|resolution |schedule|mAP (bbox/segm)|Time |
| - | - | - | - | - | - | | - | - | - | - | - | - |
|R50 |64 |(600, 1024)|280k |33.0 |22h on 8 P100| |R-50 |64 |(600, 1024)|280k |33.0 |22h on 8 P100|
|R50 |512 |(800, 1333)|280k |35.6 |55h on 8 P100| |R-50 |512 |(800, 1333)|280k |35.6 |55h on 8 P100|
|R50* |512 |(800, 1333)|360k |36.7 |49h on 8 V100| |R-50 |512 |(800, 1333)|360k |36.7 |49h on 8 V100|
|R50 |256 |(800, 1333)|280k |36.9/32.3 |39h on 8 P100| |R-50 |256 |(800, 1333)|280k |36.9/32.3 |39h on 8 P100|
|R101 |512 |(800, 1333)|280k |40.1/34.4 |70h on 8 P100| |R-50 |512 |(800, 1333)|360k |37.7/33.0 |72h on 8 P100|
|R-101 |512 |(800, 1333)|280k |40.1/34.4 |70h on 8 P100|
These models are trained with different configurations. The two 360k models have identical configurations with
The starred (*) models have identical configurations with
`R50-C4-2x` configuration in `R50-C4-2x` configuration in
[Detectron Model Zoo](https://github.com/facebookresearch/Detectron/blob/master/MODEL_ZOO.md#end-to-end-faster--mask-r-cnn-baselines0) [Detectron Model Zoo](https://github.com/facebookresearch/Detectron/blob/master/MODEL_ZOO.md#end-to-end-faster--mask-r-cnn-baselines0).
and get the same performance. They get the __same performance__ with the official models, and are about 14% slower than the official implementation, due to the lack of specialized ops.
## Notes ## Notes
......
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