Getting mIOU metric

[ghost]

I have my mIOU function, which requires the ground truth image and the binary predicted image of the model. I am using a custom dataset and the lightning inferencer. For this, I am trying to get the image with the anomalous binary mask so that I input this in my mIOU function. I do not know how to get this binary mask image from the repo. I have tried to output the pred_mask variable, which seems to be the binary mask but do not get anything at the output. What can I do to get the binary mask only?