Skip to content
Snippets Groups Projects
Unverified Commit 6e48d281 authored by Kai Chen's avatar Kai Chen Committed by GitHub
Browse files

Update readme (#1161)

* update readme

* add different datasets on model zoo
parent 08a11c17
No related branches found
No related tags found
No related merge requests found
...@@ -189,18 +189,11 @@ More models with different backbones will be added to the model zoo. ...@@ -189,18 +189,11 @@ More models with different backbones will be added to the model zoo.
| VGG16 | 300 | caffe | 120e | 3.5 | 0.256 | 25.9 / 34.6 | 25.7 | [model](https://s3.ap-northeast-2.amazonaws.com/open-mmlab/mmdetection/models/ssd300_coco_vgg16_caffe_120e_20181221-84d7110b.pth) | | VGG16 | 300 | caffe | 120e | 3.5 | 0.256 | 25.9 / 34.6 | 25.7 | [model](https://s3.ap-northeast-2.amazonaws.com/open-mmlab/mmdetection/models/ssd300_coco_vgg16_caffe_120e_20181221-84d7110b.pth) |
| VGG16 | 512 | caffe | 120e | 7.6 | 0.412 | 20.7 / 25.4 | 29.3 | [model](https://s3.ap-northeast-2.amazonaws.com/open-mmlab/mmdetection/models/ssd512_coco_vgg16_caffe_120e_20181221-d48b0be8.pth) | | VGG16 | 512 | caffe | 120e | 7.6 | 0.412 | 20.7 / 25.4 | 29.3 | [model](https://s3.ap-northeast-2.amazonaws.com/open-mmlab/mmdetection/models/ssd512_coco_vgg16_caffe_120e_20181221-d48b0be8.pth) |
### SSD (PASCAL VOC)
| Backbone | Size | Style | Lr schd | Mem (GB) | Train time (s/iter) | Inf time (fps) | box AP | Download |
| :------: | :---: | :---: | :-----: | :------: | :-----------------: | :------------: | :----: | :------------------------------------------------------------------------------------------------------------------------------: |
| VGG16 | 300 | caffe | 240e | 2.5 | 0.159 | 35.7 / 53.6 | 77.5 | [model](https://s3.ap-northeast-2.amazonaws.com/open-mmlab/mmdetection/models/ssd300_voc_vgg16_caffe_240e_20190501-7160d09a.pth) |
| VGG16 | 512 | caffe | 240e | 4.3 | 0.214 | 27.5 / 35.9 | 80.0 | [model](https://s3.ap-northeast-2.amazonaws.com/open-mmlab/mmdetection/models/ssd512_voc_vgg16_caffe_240e_20190501-ff194be1.pth) |
**Notes:** **Notes:**
- `cudnn.benchmark` is set as `True` for SSD training and testing. - `cudnn.benchmark` is set as `True` for SSD training and testing.
- Inference time is reported for batch size = 1 and batch size = 8. - Inference time is reported for batch size = 1 and batch size = 8.
- The speed difference between VOC and COCO is caused by model parameters and nms. - The speed on COCO and VOC are different due to model parameters and nms.
### Group Normalization (GN) ### Group Normalization (GN)
...@@ -222,6 +215,38 @@ Please refer to [Libra R-CNN](configs/libra_rcnn/README.md) for details. ...@@ -222,6 +215,38 @@ Please refer to [Libra R-CNN](configs/libra_rcnn/README.md) for details.
Please refer to [Guided Anchoring](configs/guided_anchoring/README.md) for details. Please refer to [Guided Anchoring](configs/guided_anchoring/README.md) for details.
### FCOS
Please refer to [FCOS](configs/fcos/README.md) for details.
### Grid R-CNN (plus)
Please refer to [Grid R-CNN](configs/grid_rcnn/README.md) for details.
### GHM
Please refer to [GHM](configs/ghm/README.md) for details.
### GCNet
Please refer to [GCNet](configs/gcnet/README.md) for details.
### HRNet
Please refer to [HRNet](configs/hrnet/README.md) for details.
### Mask Scoring R-CNN
Please refer to [Mask Scoring R-CNN](configs/ms_rcnn/README.md) for details.
### Train from Scratch
Please refer to [Rethinking ImageNet Pre-training](configs/scratch/README.md) for details.
### Other datasets
We also benchmark some methods on [PASCAL VOC](configs/pascal_voc/README.md), [Cityscapes](configs/cityscapes/README.md) and [WIDER FACE](configs/wider_face/README.md).
## Comparison with Detectron and maskrcnn-benchmark ## Comparison with Detectron and maskrcnn-benchmark
We compare mmdetection with [Detectron](https://github.com/facebookresearch/Detectron) We compare mmdetection with [Detectron](https://github.com/facebookresearch/Detectron)
......
...@@ -100,6 +100,7 @@ Results and models are available in the [Model zoo](MODEL_ZOO.md). ...@@ -100,6 +100,7 @@ Results and models are available in the [Model zoo](MODEL_ZOO.md).
| GHM | ✓ | ✓ | ☐ | ✗ | ✓ | | GHM | ✓ | ✓ | ☐ | ✗ | ✓ |
| Mask Scoring R-CNN | ✓ | ✓ | ☐ | ✗ | ✓ | | Mask Scoring R-CNN | ✓ | ✓ | ☐ | ✗ | ✓ |
| FCOS | ✓ | ✓ | ☐ | ✗ | ✓ | | FCOS | ✓ | ✓ | ☐ | ✗ | ✓ |
| Double-Head R-CNN | ✓ | ✓ | ☐ | ✗ | ✓ |
| Grid R-CNN (Plus) | ✓ | ✓ | ☐ | ✗ | ✓ | | Grid R-CNN (Plus) | ✓ | ✓ | ☐ | ✗ | ✓ |
| Hybrid Task Cascade| ✓ | ✓ | ☐ | ✗ | ✓ | | Hybrid Task Cascade| ✓ | ✓ | ☐ | ✗ | ✓ |
| Libra R-CNN | ✓ | ✓ | ☐ | ✗ | ✓ | | Libra R-CNN | ✓ | ✓ | ☐ | ✗ | ✓ |
...@@ -142,12 +143,14 @@ If you use this toolbox or benchmark in your research, please cite this project. ...@@ -142,12 +143,14 @@ If you use this toolbox or benchmark in your research, please cite this project.
``` ```
@article{mmdetection, @article{mmdetection,
title = {{MMDetection}: Open MMLab Detection Toolbox and Benchmark}, title = {{MMDetection}: Open MMLab Detection Toolbox and Benchmark},
author = {Kai Chen, Jiaqi Wang, Jiangmiao Pang, Yuhang Cao, Yu Xiong, Xiaoxiao Li, author = {Chen, Kai and Wang, Jiaqi and Pang, Jiangmiao and Cao, Yuhang and
Shuyang Sun, Wansen Feng, Ziwei Liu, Jiarui Xu, Zheng Zhang, Dazhi Cheng, Xiong, Yu and Li, Xiaoxiao and Sun, Shuyang and Feng, Wansen and
Chenchen Zhu, Tianheng Cheng, Qijie Zhao, Buyu Li, Xin Lu, Rui Zhu, Yue Wu, Liu, Ziwei and Xu, Jiarui and Zhang, Zheng and Cheng, Dazhi and
Jifeng Dai, Jingdong Wang, Jianping Shi, Wanli Ouyang, Chen Change Loy, Dahua Lin}, Zhu, Chenchen and Cheng, Tianheng and Zhao, Qijie and Li, Buyu and
journal = {arXiv preprint arXiv:1906.07155}, Lu, Xin and Zhu, Rui and Wu, Yue and Dai, Jifeng and Wang, Jingdong
year = {2019} and Shi, Jianping and Ouyang, Wanli and Loy, Chen Change and Lin, Dahua},
journal= {arXiv preprint arXiv:1906.07155},
year={2019}
} }
``` ```
......
### SSD
| Backbone | Size | Style | Lr schd | Mem (GB) | Train time (s/iter) | Inf time (fps) | box AP | Download |
| :------: | :---: | :---: | :-----: | :------: | :-----------------: | :------------: | :----: | :------------------------------------------------------------------------------------------------------------------------------: |
| VGG16 | 300 | caffe | 240e | 2.5 | 0.159 | 35.7 / 53.6 | 77.5 | [model](https://s3.ap-northeast-2.amazonaws.com/open-mmlab/mmdetection/models/ssd300_voc_vgg16_caffe_240e_20190501-7160d09a.pth) |
| VGG16 | 512 | caffe | 240e | 4.3 | 0.214 | 27.5 / 35.9 | 80.0 | [model](https://s3.ap-northeast-2.amazonaws.com/open-mmlab/mmdetection/models/ssd512_voc_vgg16_caffe_240e_20190501-ff194be1.pth) |
\ No newline at end of file
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment