PolarMask: Single Shot Instance Segmentation with Polar Representation
About
In this paper, we introduce an anchor-box free and single shot instance segmentation method, which is conceptually simple, fully convolutional and can be used as a mask prediction module for instance segmentation, by easily embedding it into most off-the-shelf detection methods. Our method, termed PolarMask, formulates the instance segmentation problem as instance center classification and dense distance regression in a polar coordinate. Moreover, we propose two effective approaches to deal with sampling high-quality center examples and optimization for dense distance regression, respectively, which can significantly improve the performance and simplify the training process. Without any bells and whistles, PolarMask achieves 32.9% in mask mAP with single-model and single-scale training/testing on challenging COCO dataset. For the first time, we demonstrate a much simpler and flexible instance segmentation framework achieving competitive accuracy. We hope that the proposed PolarMask framework can serve as a fundamental and strong baseline for single shot instance segmentation tasks. Code is available at: github.com/xieenze/PolarMask.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Instance Segmentation | COCO 2017 (val) | -- | 1201 | |
| Instance Segmentation | COCO (val) | APmk30.5 | 475 | |
| Instance Segmentation | COCO (test-dev) | APM37.7 | 380 | |
| Instance Segmentation | COCO 2017 (test-dev) | AP (Overall)32.1 | 253 | |
| Human Pose Estimation | COCO keypoint (val) | AP62.7 | 23 | |
| Instance Segmentation | SBD (val) | AP@0.50 (Mask)50.11 | 22 | |
| Instance Segmentation | iShape | mmAP (Average)3.1 | 14 | |
| Instance Segmentation | iSAID 1.0 (val) | AP27.2 | 13 | |
| Polygon Detection | MS COCO (test-dev) | mAP31.3 | 10 | |
| Instance Segmentation | COCO 36 (test-dev) | AP32.1 | 9 |