Xinlong Wang, Xiaosong Zhang, Yue Cao, Wen Wang, Chunhua Shen, Tiejun Huang
We present SegGPT, a generalist model for segmenting everything in context. We unify various segmentation tasks into a generalist in-context learning framework that accommodates different kinds of segmentation data by transforming them into the same format of images. The training of SegGPT is formulated as an in-context coloring problem with random color mapping for each data sample. The objective is to accomplish diverse tasks according to the context, rather than relying on specific colors. After training, SegGPT can perform arbitrary segmentation tasks in images or videos via in-context inference, such as object instance, stuff, part, contour, and text. SegGPT is evaluated on a broad range of tasks, including few-shot semantic segmentation, video object segmentation, semantic segmentation, and panoptic segmentation. Our results show strong capabilities in segmenting in-domain and out-of-domain targets, either qualitatively or quantitatively.
| Task | Dataset | Metric | Value | Model |
|---|---|---|---|---|
| Few-Shot Learning | FSS-1000 (5-shot) | Mean IoU | 89.3 | SegGPT (ViT) |
| Few-Shot Learning | COCO-20i (5-shot) | Mean IoU | 67.9 | SegGPT (ViT) |
| Few-Shot Learning | FSS-1000 (1-shot) | Mean IoU | 85.6 | SegGPT (ViT) |
| Few-Shot Learning | PASCAL-5i (1-Shot) | Mean IoU | 83.2 | SegGPT (ViT) |
| Few-Shot Learning | COCO-20i (1-shot) | Mean IoU | 56.1 | SegGPT (ViT) |
| Few-Shot Learning | PASCAL-5i (5-Shot) | Mean IoU | 89.8 | SegGPT (ViT) |
| Few-Shot Semantic Segmentation | FSS-1000 (5-shot) | Mean IoU | 89.3 | SegGPT (ViT) |
| Few-Shot Semantic Segmentation | COCO-20i (5-shot) | Mean IoU | 67.9 | SegGPT (ViT) |
| Few-Shot Semantic Segmentation | FSS-1000 (1-shot) | Mean IoU | 85.6 | SegGPT (ViT) |
| Few-Shot Semantic Segmentation | PASCAL-5i (1-Shot) | Mean IoU | 83.2 | SegGPT (ViT) |
| Few-Shot Semantic Segmentation | COCO-20i (1-shot) | Mean IoU | 56.1 | SegGPT (ViT) |
| Few-Shot Semantic Segmentation | PASCAL-5i (5-Shot) | Mean IoU | 89.8 | SegGPT (ViT) |
| Meta-Learning | FSS-1000 (5-shot) | Mean IoU | 89.3 | SegGPT (ViT) |
| Meta-Learning | COCO-20i (5-shot) | Mean IoU | 67.9 | SegGPT (ViT) |
| Meta-Learning | FSS-1000 (1-shot) | Mean IoU | 85.6 | SegGPT (ViT) |
| Meta-Learning | PASCAL-5i (1-Shot) | Mean IoU | 83.2 | SegGPT (ViT) |
| Meta-Learning | COCO-20i (1-shot) | Mean IoU | 56.1 | SegGPT (ViT) |
| Meta-Learning | PASCAL-5i (5-Shot) | Mean IoU | 89.8 | SegGPT (ViT) |