arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.03546cs.CVcs.CLcs.LG

Language-driven Semantic Segmentation

Cornell University(康奈尔大学);Cornell Tech(康奈尔科技学院);University of Copenhagen(哥本哈根大学);Apple(苹果公司);Intel Labs(英特尔实验室)

Boyi Li, Kilian Q. Weinberger, Serge Belongie, Vladlen Koltun, René Ranftl

英文摘要:

We present LSeg, a novel model for language-driven semantic image segmentation. LSeg uses a text encoder to compute embeddings of descriptive input labels (e.g., "grass" or "building") together with a transformer-based image encoder that computes dense per-pixel embeddings of the input image. The image encoder is trained with a contrastive objective to align pixel embeddings to the text embedding of the corresponding semantic class. The text embeddings provide a flexible label representation in which semantically similar labels map to similar regions in the embedding space (e.g., "cat" and "furry"). This allows LSeg to generalize to previously unseen categories at test time, without retraining or even requiring a single additional training sample. We demonstrate that our approach achieves highly competitive zero-shot performance compared to existing zero- and few-shot semantic segmentation methods, and even matches the accuracy of traditional segmentation algorithms when a fixed label set is provided. Code and demo are available at https://github.com/isl-org/lang-seg.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑