README.md

Cascade R-CNN: Delving into High Quality Object Detection

by Zhaowei Cai and Nuno Vasconcelos

This repository is written by Zhaowei Cai at UC San Diego, on the base of Detectron @ e8942c8.

Introduction

This repository re-implements Cascade R-CNN on the base of Detectron. Very consistent gains are available for all tested models, regardless of baseline strength.

This repository is still under construction. Will be updated very soon!

Citation

If you use our code/model/data, please cite our paper:

@inproceedings{cai18cascadercnn,
  author = {Zhaowei Cai and Nuno Vasconcelos},
  Title = {Cascade R-CNN: Delving into High Quality Object Detection},
  booktitle = {CVPR},
  Year  = {2018}
}

and Detectron:

@misc{Detectron2018,
  author =       {Ross Girshick and Ilija Radosavovic and Georgia Gkioxari and
                  Piotr Doll\'{a}r and Kaiming He},
  title =        {Detectron},
  howpublished = {\url{https://github.com/facebookresearch/detectron}},
  year =         {2018}
}

Benchmarking

End-to-End Faster & Mask R-CNN Baselines

        backbone         type lr
schd im/
gpu box
AP box
AP50 box
AP75 mask
AP mask
AP50 mask
AP75 download
links

R-50-FPN-baseline Faster 1x 2 36.7 - - - - - model | boxes R-50-FPN-cascade Faster 1x 2 40.9 59.0 44.6 - - - model | boxes R-101-FPN-baseline Faster 1x 2 39.4 - - - - - model | boxes R-101-FPN-cascade Faster 1x 2 42.8 61.4 46.1 - - - model | boxes X-101-64x4d-FPN-baseline Faster 1x 1 41.5 - - - - - model | boxes X-101-64x4d-FPN-cascade Faster 1x 1 45.4 64.0 49.8 - - - model | boxes X-101-32x8d-FPN-baseline Faster 1x 1 41.3 - - - - - model | boxes X-101-32x8d-FPN-cascade Faster 1x 1 44.7 63.7 48.8 - - - model | boxes R-50-FPN-baseline Mask 1x 2 37.7 - - 33.9 - - model | boxes | masks R-50-FPN-cascade Mask 1x 2 41.3 59.6 44.9 35.4 56.2 37.8 model | boxes | masks R-101-FPN-baseline Mask 1x 2 40.0 - - 35.9 - - model | boxes | masks R-101-FPN-cascade Mask 1x 2 - - - - - - model | boxes | masks X-101-64x4d-FPN-baseline Mask 1x 1 42.4 - - 37.5 - - model | boxes | masks X-101-64x4d-FPN-cascade Mask 1x 1 45.9 64.4 50.2 38.8 61.3 41.6 model | boxes | masks X-101-32x8d-FPN-baseline Mask 1x 1 42.1 - - 37.3 - - model | boxes | masks X-101-32x8d-FPN-cascade Mask 1x 1 45.8 64.1 50.3 38.6 60.6 41.5 model | boxes | masks

Mask R-CNN with Bells & Whistles

        backbone         type lr
schd im/
gpu box
AP box
AP50 box
AP75 mask
AP mask
AP50 mask
AP75 download
links

X-152-32x8d-FPN-IN5k-baseline Mask s1x 1 48.1 - - 41.5 - - model | boxes | masks [above without test-time aug.]

45.2 - - 39.7 - -

X-152-32x8d-FPN-IN5k-cascade Mask s1x 1 50.2 68.2 55.0 42.3 65.4 45.8 model | boxes | masks [above without test-time aug.]

48.1 66.7 52.6 40.7 63.7 43.8