README.md 7.5 KB

FCOS

FCOS: Fully Convolutional One-Stage Object Detection

Abstract

We propose a fully convolutional one-stage object detector (FCOS) to solve object detection in a per-pixel prediction fashion, analogue to semantic segmentation. Almost all state-of-the-art object detectors such as RetinaNet, SSD, YOLOv3, and Faster R-CNN rely on pre-defined anchor boxes. In contrast, our proposed detector FCOS is anchor box free, as well as proposal free. By eliminating the predefined set of anchor boxes, FCOS completely avoids the complicated computation related to anchor boxes such as calculating overlapping during training. More importantly, we also avoid all hyper-parameters related to anchor boxes, which are often very sensitive to the final detection performance. With the only post-processing non-maximum suppression (NMS), FCOS with ResNeXt-64x4d-101 achieves 44.7% in AP with single-model and single-scale testing, surpassing previous one-stage detectors with the advantage of being much simpler. For the first time, we demonstrate a much simpler and flexible detection framework achieving improved detection accuracy. We hope that the proposed FCOS framework can serve as a simple and strong alternative for many other instance-level tasks.

Results and Models

Backbone Style GN MS train Tricks DCN Lr schd Mem (GB) Inf time (fps) box AP Config Download
R-50 caffe Y N N N 1x 3.6 22.7 36.6 config model | log
R-50 caffe Y N Y N 1x 3.7 - 38.7 config model | log
R-50 caffe Y N Y Y 1x 3.8 - 42.3 config model | log
R-101 caffe Y N N N 1x 5.5 17.3 39.1 config model | log
Backbone Style GN MS train Lr schd Mem (GB) Inf time (fps) box AP Config Download
R-50 caffe Y Y 2x 2.6 22.9 38.5 config model | log
R-101 caffe Y Y 2x 5.5 17.3 40.8 config model | log
X-101 pytorch Y Y 2x 10.0 9.7 42.6 config model | log

Notes:

  • The X-101 backbone is X-101-64x4d.
  • Tricks means setting norm_on_bbox, centerness_on_reg, center_sampling as True.
  • DCN means using DCNv2 in both backbone and head.

Citation

@article{tian2019fcos,
  title={FCOS: Fully Convolutional One-Stage Object Detection},
  author={Tian, Zhi and Shen, Chunhua and Chen, Hao and He, Tong},
  journal={arXiv preprint arXiv:1904.01355},
  year={2019}
}