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

275 stars 36 forks source link

Open chinmay5 opened 1 month ago

chinmay5 commented 1 month ago

Hi,

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

Best, Chinmay