SegGPT

SegGPT: Segmenting Everything In Context

Xinlong Wang^1*,   Xiaosong Zhang^1*,   Yue Cao^1*,   Wen Wang²,   Chunhua Shen²,   Tiejun Huang^1,3

¹BAAI,   ²ZJU,   ³PKU

Enjoy the Demo

We present SegGPT, a generalist model for segmenting everything in context. With only one single model, 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.

[Paper] [Demo]

Run the demo

• We provide a UI with gradio for running the demo locally. Running the following command in a terminal will launch the demo:

  python app_gradio.py
  

• We combined SAM and SegGPT to enable one touch for segmentation in all images (一触百通), as well as any segmentation in a video.
• This demo is also hosted on HuggingFace here.

Inference

• We provide the model and code for SegGPT inference, please refer to SegGPT_inference/README.md.

Citation

@article{SegGPT,
  title={SegGPT: Segmenting Everything In Context},
  author={Wang, Xinlong and Zhang, Xiaosong and Cao, Yue and Wang, Wen and Shen, Chunhua and Huang, Tiejun},
  journal={arXiv preprint arXiv:2304.03284},
  year={2023}
}

Contact

We are hiring at all levels at BAAI Vision Team, including full-time researchers, engineers and interns. If you are interested in working with us on foundation model, visual perception and multimodal learning, please contact Xinlong Wang (wangxinlong@baai.ac.cn) and Yue Cao (caoyue@baai.ac.cn).