实例分割中mask的IoU的代码分析)
(话不多说上代码)
def mask_iou(mask1, mask2):
"""
mask1: [m1,n] m1 means number of predicted objects
mask2: [m2,n] m2 means number of gt objects
Note: n means image_w x image_h
"""
intersection = torch.matmul(mask1, mask2.t())
area1 = torch.sum(mask1, dim=1).view(1, -