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 produces state-of-the-art results on the Cityscapes validation set, the Vaihingen building segmentation benchmark, and the MoNuSeg dataset.
updated: Mon Sep 05 2022 04:54:36 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アソシエイト