search

ge-xing / Diff-UNet

Diff-UNet: A Diffusion Embedded Network for Volumetric Segmentation. (using diffusion for 3D medical image segmentation)
Apache License 2.0
141 stars 20 forks source link

Can I use this model on verse2020? #17

Open XFivezzz opened 1 year ago

920232796 commented 1 year ago

Yes

XFivezzz commented 1 year ago

Thank you for your reply.I think the code integration is too strong. Can you give me some suggestions on how to apply this model to verse2020 data set

920232796 commented 1 year ago

You need to modify the dataset class to get images and labels from the verse2020 dataset. Then you need to modify the parameters of the Diff-UNet.