iou.py

 
def iou(img, y, c):
    intersection = 0.
    union = 0.
    img = img.reshape(-1)
    y = y.reshape(-1)
    for i in range(len(img)):
         intersection += img[i] == c and y[i] == c
         union += img[i] == c or y[i] == c
    return intersection / union