Example of performing multi-class segmentation?

JuliaWolleb / Diffusion-based-Segmentation

This is the official Pytorch implementation of the paper "Diffusion Models for Implicit Image Segmentation Ensembles".

MIT License

284 stars 38 forks source link

Open chinmay5 opened 4 months ago

chinmay5 commented 4 months ago

Hi,

Can you please tell me the steps to update the script to perform a multi-class segmentation on the inputs?

Best, Chinmay