code: def iou(mask_a, mask_b, class_id):
      return np.nan_to_num(
          ((mask_a == class_id) & (mask_b == class_id)).sum(axis=(1, 2))
          / ((mask_a == class_id) | (mask_b == class_id)).sum(
              axis=(1, 2)
          ),
          0,
          0,
          0,
      )
  
imports: import os
  import pickle
  import shutil
  import time
  import numpy
  import wandb
  from PIL import Image
label_desc: Use image node_id 15 for calling a global function with example usage: iou(mask_a, mask_b, class_id) and returns: np