arXiv reaDer
SegDiff:拡散確率モデルによる画像セグメンテーション
SegDiff: Image Segmentation with Diffusion Probabilistic Models
拡散確率的手法は、最先端の画像生成に採用されています。この作業では、画像セグメンテーションを実行するためにそのようなモデルを拡張する方法を提示します。このメソッドは、事前にトレーニングされたバックボーンに依存することなく、エンドツーエンドで学習します。入力画像とセグメンテーションマップの現在の推定値の情報は、2つのエンコーダーの出力を合計することによってマージされます。次に、追加のエンコーディングレイヤーとデコーダーを使用して、拡散モデルを使用してセグメンテーションマップを繰り返し改良します。拡散モデルは確率的であるため、複数回適用され、結果が最終的なセグメンテーションマップにマージされます。新しいメソッドは、Cityscapes検証セット、Vaihingen建物セグメンテーションベンチマーク、およびMoNuSegデータセットに関する最新の結果を取得します。
Diffusion Probabilistic Methods are employed for state-of-the-art image generation. In this work, we present a method for extending such models for performing image segmentation. The method learns end-to-end, without relying on a pre-trained backbone. The information in the input image and in the current estimation of the segmentation map is merged by summing the output of two encoders. Additional encoding layers and a decoder are then used to iteratively refine the segmentation map using a diffusion model. Since the diffusion model is probabilistic, it is applied multiple times and the results are merged into a final segmentation map. The new method obtains state-of-the-art results on the Cityscapes validation set, the Vaihingen building segmentation benchmark, and the MoNuSeg dataset.
updated: Wed Dec 01 2021 10:17:25 GMT+0000 (UTC)
published: Wed Dec 01 2021 10:17:25 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アソシエイト