arXiv reaDer
言語主導のセマンティックセグメンテーション
Language-driven Semantic Segmentation
言語駆動型セマンティック画像セグメンテーションの新しいモデルであるLSegを紹介します。 LSegは、テキストエンコーダーを使用して、入力画像のピクセルごとの高密度埋め込みを計算するトランスフォーマーベースの画像エンコーダーとともに、説明的な入力ラベル(「草」や「建物」など)の埋め込みを計算します。画像エンコーダーは、対応するセマンティッククラスのテキスト埋め込みにピクセル埋め込みを揃えるために、対照的な目的でトレーニングされています。テキスト埋め込みは、意味的に類似したラベルが埋め込みスペース内の類似した領域(たとえば、「猫」や「毛皮」)にマッピングされる柔軟なラベル表現を提供します。これにより、LSegは、再トレーニングしたり、追加のトレーニングサンプルを1つも必要とせずに、テスト時に以前は表示されなかったカテゴリに一般化できます。私たちのアプローチは、既存のゼロショットおよび数ショットのセマンティックセグメンテーション方法と比較して非常に競争力のあるゼロショットパフォーマンスを達成し、固定ラベルセットが提供されている場合は従来のセグメンテーションアルゴリズムの精度にも匹敵することを示します。コードとデモはhttps://github.com/isl-org/lang-segで入手できます。
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.
updated: Mon Jan 10 2022 18:59:10 GMT+0000 (UTC)
published: Mon Jan 10 2022 18:59:10 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト