Image Segmentation Using Text and Image Prompts

By Timo Lüddecke, Alexander S. Ecker. 2021

794 citations recorded by the source corpus.

Image segmentation is usually addressed by training a model for a fixed set of object classes. Incorporating additional classes or more complex queries later is expensive as it requires re-training the model on a dataset that encompasses these expressions. Here we propose a system that can generate image segmentations based on arbitrary prompts at test time. A prompt can be either a text or an image. This approach enables us to create a unified model (trained once) for three common segmentation tasks, which come with distinct challenges: referring expression segmentation, zero-shot segmentation and one-shot segmentation. We build upon the CLIP model as a backbone which we extend with a transformer-based decoder that enables dense prediction. After training on an extended version of the PhraseCut dataset, our system generates a binary segmentation map for an image based on a free-text prompt or on an additional image expressing the query. We analyze different variants of the latter image-based prompts in detail. This novel hybrid input allows for dynamic adaptation not only to the three segmentation tasks mentioned above, but to any binary segmentation task where a text or image query can be formulated. Finally, we find our system to adapt well to generalized queries involving affordances or properties. Code is available at https://eckerlab.org/code/CLIPSeg

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Multimodal Learning

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Jianfeng Gao
  2. Zhe Gan
  3. Chunyuan Li
  4. Ziwei Liu
  5. Jianwei Yang
  6. Mohit Bansal
  7. Lijuan Wang
  8. Yong Jae Lee

In Computer Vision and Pattern Recognition

  1. Luc Van Gool
  2. Ming-Hsuan Yang
  3. Dacheng Tao
  4. Xiaogang Wang
  5. Trevor Darrell
  6. Chen Change Loy
  7. Cordelia Schmid
  8. Andrea Vedaldi

In Multimodal Learning

  1. Zhe Gan
  2. Jianfeng Gao
  3. Chunyuan Li
  4. Lijuan Wang
  5. Linjie Li
  6. Jianfeng Wang
  7. Jianwei Yang
  8. Yong Jae Lee

Citations in the corpus