JuliaWolleb / Diffusion-based-Segmentation

This is the official Pytorch implementation of the paper "Diffusion Models for Implicit Image Segmentation Ensembles".
MIT License
271 stars 35 forks source link

About predict_xstart settings #37

Closed GYDDHPY closed 1 year ago

GYDDHPY commented 1 year ago

Hi, thank you for your great contribution!

I have a silly question. It seems you set predict_xstart as False, but the segmentation model is used to predict label mask. Should we set predict_xstart as True? Will this setting affect the final results?

Looking forward to your reply!

GYDDHPY commented 1 year ago

Sorry to bother you. I figure out it.