68e9addae4d913c82de5696ee950619c743118c8,torch_geometric/utils/metric.py,,mean_iou,#Any#Any#Any#Any#,149

Before Change

    :rtype: :class:`Tensor`
    
    pred, target = F.one_hot(pred, num_classes), F.one_hot(target, num_classes)

    if batch is not None:
        i = scatter_add(pred & target, batch, dim=0).to(torch.float)
        u = scatter_add(pred | target, batch, dim=0).to(torch.float)
    else:
        i = (pred & target).sum(dim=0).to(torch.float)
        u = (pred | target).sum(dim=0).to(torch.float)

    iou = i / u
    iou[torch.isnan(iou)] = 1
    iou = iou.mean(dim=-1)
    return iou

After Change

    :rtype: :class:`Tensor`
    
    i, u = intersection_and_union(pred, target, num_classes, batch)
    iou = i.to(torch.float) / u.to(torch.float)
    iou[torch.isnan(iou)] = 1
    iou = iou.mean(dim=-1)

Instances