diff --git a/MODEL_ZOO.md b/MODEL_ZOO.md index 52dc1513e4e47403190d2dfd31daf9194dae5754..3d2eaae6bf4d822b29b1473067f4dd92d18d9ed3 100644 --- a/MODEL_ZOO.md +++ b/MODEL_ZOO.md @@ -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 | 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:** - `cudnn.benchmark` is set as `True` for SSD training and testing. - 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) @@ -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. +### 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 We compare mmdetection with [Detectron](https://github.com/facebookresearch/Detectron) diff --git a/README.md b/README.md index 6b20f8654f8d317b8ec88fe1201560d8d4c82248..9aaeb4af6a42113b7e6bf325f241fa52b86bce7e 100644 --- a/README.md +++ b/README.md @@ -100,6 +100,7 @@ Results and models are available in the [Model zoo](MODEL_ZOO.md). | GHM | ✓ | ✓ | ☠| ✗ | ✓ | | Mask Scoring R-CNN | ✓ | ✓ | ☠| ✗ | ✓ | | FCOS | ✓ | ✓ | ☠| ✗ | ✓ | +| Double-Head R-CNN | ✓ | ✓ | ☠| ✗ | ✓ | | Grid R-CNN (Plus) | ✓ | ✓ | ☠| ✗ | ✓ | | Hybrid Task Cascade| ✓ | ✓ | ☠| ✗ | ✓ | | Libra R-CNN | ✓ | ✓ | ☠| ✗ | ✓ | @@ -142,12 +143,14 @@ If you use this toolbox or benchmark in your research, please cite this project. ``` @article{mmdetection, title = {{MMDetection}: Open MMLab Detection Toolbox and Benchmark}, - author = {Kai Chen, Jiaqi Wang, Jiangmiao Pang, Yuhang Cao, Yu Xiong, Xiaoxiao Li, - Shuyang Sun, Wansen Feng, Ziwei Liu, Jiarui Xu, Zheng Zhang, Dazhi Cheng, - Chenchen Zhu, Tianheng Cheng, Qijie Zhao, Buyu Li, Xin Lu, Rui Zhu, Yue Wu, - Jifeng Dai, Jingdong Wang, Jianping Shi, Wanli Ouyang, Chen Change Loy, Dahua Lin}, - journal = {arXiv preprint arXiv:1906.07155}, - year = {2019} + author = {Chen, Kai and Wang, Jiaqi and Pang, Jiangmiao and Cao, Yuhang and + Xiong, Yu and Li, Xiaoxiao and Sun, Shuyang and Feng, Wansen and + Liu, Ziwei and Xu, Jiarui and Zhang, Zheng and Cheng, Dazhi and + Zhu, Chenchen and Cheng, Tianheng and Zhao, Qijie and Li, Buyu and + Lu, Xin and Zhu, Rui and Wu, Yue and Dai, Jifeng and Wang, Jingdong + and Shi, Jianping and Ouyang, Wanli and Loy, Chen Change and Lin, Dahua}, + journal= {arXiv preprint arXiv:1906.07155}, + year={2019} } ``` diff --git a/configs/pascal_voc/README.md b/configs/pascal_voc/README.md new file mode 100644 index 0000000000000000000000000000000000000000..a317dfd9a438b0851a623a75840826c47c41e9d6 --- /dev/null +++ b/configs/pascal_voc/README.md @@ -0,0 +1,6 @@ +### 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