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

update maskrcnn notes

parent dc13638b
......@@ -25,8 +25,8 @@ DIR/
## Usage
Change config in `config.py`:
1. Set `BASEDIR` to `/path/to/DIR` as described above.
2. Set `MODE_MASK` to switch Faster-RCNN or Mask-RCNN.
1. Change `BASEDIR` to `/path/to/DIR` as described above.
2. Change `MODE_MASK` to switch Faster-RCNN or Mask-RCNN.
Train:
```
......@@ -48,22 +48,22 @@ Evaluate the performance of a model and save to json.
## 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.
|Backbone|`FASTRCNN_BATCH`|resolution |schedule|mAP (bbox/segm)|Time |
| - | - | - | - | - | - |
|R50 |64 |(600, 1024)|280k |33.0 |22h on 8 P100|
|R50 |512 |(800, 1333)|280k |35.6 |55h on 8 P100|
|R50* |512 |(800, 1333)|360k |36.7 |49h on 8 V100|
|R50 |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 |64 |(600, 1024)|280k |33.0 |22h on 8 P100|
|R-50 |512 |(800, 1333)|280k |35.6 |55h on 8 P100|
|R-50 |512 |(800, 1333)|360k |36.7 |49h on 8 V100|
|R-50 |256 |(800, 1333)|280k |36.9/32.3 |39h 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 starred (*) models have identical configurations with
The two 360k models have identical configurations with
`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)
and get the same performance.
[Detectron Model Zoo](https://github.com/facebookresearch/Detectron/blob/master/MODEL_ZOO.md#end-to-end-faster--mask-r-cnn-baselines0).
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
......
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