How to generate a heatmap of inference output probabilities?

[hadign20]

Hi

Please let me know how can I generate a heatmap of tumor probabilities alongside the final thresholded segmentations?

Thank you