1. SSAP: Single-Shot Instance Segmentation With Affinity Pyramid
- Author
-
Kaiqi Huang, Yanhu Shan, Xin Zhao, Yupei Wang, and Naiyu Gao
- Subjects
Pixel ,business.industry ,Computer science ,Graph partition ,Pattern recognition ,02 engineering and technology ,Image segmentation ,Object (computer science) ,Pyramid ,0202 electrical engineering, electronic engineering, information engineering ,Media Technology ,020201 artificial intelligence & image processing ,Segmentation ,Artificial intelligence ,Pyramid (image processing) ,Electrical and Electronic Engineering ,business - Abstract
Proposal-free instance segmentation methods mainly generate instance-agnostic semantic segmentation labels and instance-aware features to group pixels into different object instances. However, previous methods mostly employ separate modules for these two sub-tasks and require multiple passes for inference. In addition to the lack of efficiency, previous methods also failed to perform as well as proposal-based approaches. To this end, this work proposes a single-shot proposal-free instance segmentation method that requires only one single pass for prediction. Our method is based on learning an affinity pyramid, which computes the probability that two pixels belong to the same instance in a hierarchical manner. Moreover, incorporating with the learned affinity pyramid, a novel cascaded graph partition (CGP) module is presented to fuse the two predictions and segment instances efficiently. As an additional contribution, we conduct an experiment to demonstrate the benefits of proposal-free methods in capturing detailed structures from finely annotated training examples. Our approach is evaluated on the Cityscapes and COCO datasets and achieves state-of-the-art performance.
- Published
- 2021
- Full Text
- View/download PDF