denseCRF output

Hi there,

Thank you for your code.
I've tried denseCRF for multi-class segmentation problem but I just found that this method will output class labels directly. Just wondering if it's possible to output the probability of each class, like what we input (N, C, H, W).

Thank you!