OpenMMLab Detection Toolbox and Benchmark
https://mmdetection.readthedocs.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
3.8 KiB
3.8 KiB
Feature Pyramid Grids
Introduction
@article{chen2020feature,
title={Feature pyramid grids},
author={Chen, Kai and Cao, Yuhang and Loy, Chen Change and Lin, Dahua and Feichtenhofer, Christoph},
journal={arXiv preprint arXiv:2004.03580},
year={2020}
}
Results and Models
We benchmark the new training schedule (crop training, large batch, unfrozen BN, 50 epochs) introduced in NAS-FPN. All backbones are Resnet-50 in pytorch style.
| Method | Neck | Lr schd | Mem (GB) | Inf time (fps) | box AP | mask AP | Config | Download |
|---|---|---|---|---|---|---|---|---|
| Faster R-CNN | FPG | 50e | 20.0 | - | 42.2 | - | config | model | log |
| Faster R-CNN | FPG-chn128 | 50e | 11.9 | - | 41.2 | - | config | model | log |
| Mask R-CNN | FPG | 50e | 23.2 | - | 42.7 | 37.8 | config | model | log |
| Mask R-CNN | FPG-chn128 | 50e | 15.3 | - | 41.7 | 36.9 | config | model | log |
| RetinaNet | FPG | 50e | 20.8 | - | 40.5 | - | config | model | log |
| RetinaNet | FPG-chn128 | 50e | 19.9 | - | 40.3 | - | config | model | log |
Note: Chn128 means to decrease the number of channels of features and convs from 256 (default) to 128 in Neck and BBox Head, which can greatly decrease memory consumption without sacrificing much precision.