Trying to calculate Mean IoU for Semantic Segmentation Task

Question

I am trying to calculate loss and mean IoU for semantic segmentation task. I created a dummy variable, seg_gt_cl5, which shows the ground truth val