search

lucidrains / med-seg-diff-pytorch

Implementation of MedSegDiff in Pytorch - SOTA medical segmentation using DDPM and filtering of features in fourier space
MIT License
209 stars 28 forks source link

Mask question #14

Closed Alan-Py closed 1 year ago

Alan-Py commented 1 year ago

My input mask for example is 1 0 0 0 1 1 1 2 0

0 is background 1 is class 1 2 is class 2

Can it use in this model,or I must use one-hot mask?

Alan-Py commented 1 year ago

My input mask for example is 1 0 0 0 1 1 1 2 0

0 is background 1 is class 1 2 is class 2

Can it use in this model,or I must use one-hot mask?

we can use Ordinal Encoding for mask.