Using CRF as postprocessing step with MONAI UNetr model

[soumbane]

Hi,

I wanted to use the CRF module of MONAI as a postprocessing step with a MONAI UNETR model. Is there any example for this? How do I use the CRF model after the UNETR model output?

Like, what is the reference tensor? Is it the ground-truth labels one-hot encoded or not? And is the input tensor the model outputs without softmax?

Please help.