arXiv · 2112.00390
SegDiff: Image Segmentation with Diffusion Probabilistic Models
Abstract
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.
Explore related subjects
Keep this discovery
Tomer Amit, Tal Shaharbany, Eliya Nachmani, Lior Wolf. 2021-12-01. SegDiff: Image Segmentation with Diffusion Probabilistic Models. https://arxiv.org/abs/2112.00390
Cite the original work for its findings. Save a collection to share your selection of sources.