arXiv reaDer
テキストから画像への拡散モデルによるオープン語彙のパノプティック セグメンテーション
Open-Vocabulary Panoptic Segmentation with Text-to-Image Diffusion Models
ODISE: Open-vocabulary DIffusion-based panoptic SEgmentation を提示します。これは、事前にトレーニングされたテキスト画像拡散と識別モデルを統合して、オープン語彙のパノプティック セグメンテーションを実行します。テキストから画像への拡散モデルには、多様なオープン語彙の言語記述で高品質の画像を生成する驚くべき能力があります。これは、彼らの内部表現空間が現実世界のオープンな概念と高度に相関していることを示しています。一方、CLIP のようなテキスト画像識別モデルは、画像をオープン語彙ラベルに分類するのに適しています。これらの両方のモデルの凍結された内部表現を活用して、実際のあらゆるカテゴリのパノプティック セグメンテーションを実行します。私たちのアプローチは、オープン語彙のパノプティック タスクとセマンティック セグメンテーション タスクの両方で、以前の最先端技術よりも大幅に優れています。具体的には、COCO トレーニングのみで、私たちの方法は ADE20K データセットで 23.4 PQ と 30.0 mIoU を達成し、以前の最先端技術よりも 8.3 PQ と 7.9 mIoU の絶対的な改善が見られます。 https://github.com/NVlabs/ODISE でコードとモデルをオープンソース化しています。
We present ODISE: Open-vocabulary DIffusion-based panoptic SEgmentation, which unifies pre-trained text-image diffusion and discriminative models to perform open-vocabulary panoptic segmentation. Text-to-image diffusion models have the remarkable ability to generate high-quality images with diverse open-vocabulary language descriptions. This demonstrates that their internal representation space is highly correlated with open concepts in the real world. Text-image discriminative models like CLIP, on the other hand, are good at classifying images into open-vocabulary labels. We leverage the frozen internal representations of both these models to perform panoptic segmentation of any category in the wild. Our approach outperforms the previous state of the art by significant margins on both open-vocabulary panoptic and semantic segmentation tasks. In particular, with COCO training only, our method achieves 23.4 PQ and 30.0 mIoU on the ADE20K dataset, with 8.3 PQ and 7.9 mIoU absolute improvement over the previous state of the art. We open-source our code and models at https://github.com/NVlabs/ODISE .
updated: Wed Apr 05 2023 17:40:38 GMT+0000 (UTC)
published: Wed Mar 08 2023 18:58:26 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アソシエイト