Open-World Entity Segmentation
arXiv:2107.14228
Abstract
We introduce a new image segmentation task, called Entity Segmentation (ES), which aims to segment all visual entities (objects and stuffs) in an image without predicting their semantic labels. By removing the need of class label prediction, the models trained for such task can focus more on improving segmentation quality. It has many practical applications such as image manipulation and editing where the quality of segmentation masks is crucial but class labels are less important. We conduct the first-ever study to investigate the feasibility of convolutional center-based representation to segment things and stuffs in a unified manner, and show that such representation fits exceptionally well in the context of ES. More specifically, we propose a CondInst-like fully-convolutional architecture with two novel modules specifically designed to exploit the class-agnostic and non-overlapping requirements of ES. Experiments show that the models designed and trained for ES significantly outperforms popular class-specific panoptic segmentation models in terms of segmentation quality. Moreover, an ES model can be easily trained on a combination of multiple datasets without the need to resolve label conflicts in dataset merging, and the model trained for ES on one or more datasets can generalize very well to other test datasets of unseen domains. The code has been released at https://github.com/dvlab-research/Entity/.
Project page: http://luqi.info/Entity_Web
References in corpus (6)
- Rethinking Atrous Convolution for Semantic Image Segmentation
- Swin Transformer: Hierarchical Vision Transformer using Shifted Windows
- High-Resolution Image Synthesis and Semantic Manipulation with Conditional GANs
- Image Inpainting via Generative Multi-column Convolutional Neural Networks
- FrankMocap: Fast Monocular 3D Hand and Body Motion Capture by Regression and Integration
- Sequential Context Encoding for Duplicate Removal