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}
}