Skip to content

Latest commit

 

History

History
 
 

panoptic_fpn

Panoptic FPN

Panoptic feature pyramid networks

Abstract

The recently introduced panoptic segmentation task has renewed our community's interest in unifying the tasks of instance segmentation (for thing classes) and semantic segmentation (for stuff classes). However, current state-of-the-art methods for this joint task use separate and dissimilar networks for instance and semantic segmentation, without performing any shared computation. In this work, we aim to unify these methods at the architectural level, designing a single network for both tasks. Our approach is to endow Mask R-CNN, a popular instance segmentation method, with a semantic segmentation branch using a shared Feature Pyramid Network (FPN) backbone. Surprisingly, this simple baseline not only remains effective for instance segmentation, but also yields a lightweight, top-performing method for semantic segmentation. In this work, we perform a detailed study of this minimally extended version of Mask R-CNN with FPN, which we refer to as Panoptic FPN, and show it is a robust and accurate baseline for both tasks. Given its effectiveness and conceptual simplicity, we hope our method can serve as a strong baseline and aid future research in panoptic segmentation.

Dataset

PanopticFPN requires COCO and COCO-panoptic dataset for training and evaluation. You need to download and extract it in the COCO dataset path. The directory should be like this.

mmdetection
├── mmdet
├── tools
├── configs
├── data
│   ├── coco
│   │   ├── annotations
│   │   │   ├── panoptic_train2017.json
│   │   │   ├── panoptic_train2017
│   │   │   ├── panoptic_val2017.json
│   │   │   ├── panoptic_val2017
│   │   ├── train2017
│   │   ├── val2017
│   │   ├── test2017

Results and Models

Backbone style Lr schd Mem (GB) Inf time (fps) PQ SQ RQ PQ_th SQ_th RQ_th PQ_st SQ_st RQ_st Config Download
R-50-FPN pytorch 1x 4.7 40.2 77.8 49.3 47.8 80.9 57.5 28.9 73.1 37.0 config model | log
R-50-FPN pytorch 3x - - 42.5 78.1 51.7 50.3 81.5 60.3 30.7 73.0 38.8 config model | log
R-101-FPN pytorch 1x 6.7 42.2 78.3 51.4 50.1 81.4 59.9 30.3 73.6 38.5 config model | log
R-101-FPN pytorch 3x - - 44.1 78.9 53.6 52.1 81.7 62.3 32.0 74.6 40.3 config model | log
R2-50-FPN pytorch 1x - - 42.5 78.0 51.8 50.0 81.4 60.0 31.1 72.8 39.4 config model | log

Citation

The base method for panoptic segmentation task.

@inproceedings{kirillov2018panopticfpn,
  author = {
    Alexander Kirillov,
    Ross Girshick,
    Kaiming He,
    Piotr Dollar,
  },
  title = {Panoptic Feature Pyramid Networks},
  booktitle = {Proceedings of IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
  year = {2019}
}