dfac51fcd0fbceb3f394a1d8f1ab7c8652e9cf78,gluonvision/utils/metrics/voc_segmentation.py,,batch_intersection_union,#Any#Any#Any#Any#,19

Before Change

                                 range=(mini, maxi))
    area_pred, _ = np.histogram(predict.asnumpy(), bins=nbins,
                                range=(mini, maxi))
    area_lab, _ = np.histogram(target.asnumpy(), bins=nbins,
                               range=(mini, maxi))
    area_union = area_pred + area_lab - area_inter
    return area_inter, area_union

After Change

    maxi = nclass - 1
    nbins = nclass
    predict = predict.asnumpy()
    target = target.asnumpy()
    if ignore_bg:
        mini = 1
        nbins -= 1
        predict = predict * (target > 0).astype(predict.dtype)
    else:
        predict = predict * (target >= 0).astype(predict.dtype)
    //intersection = predict * (F.equal(predict, target)).astype(predict.dtype)
    intersection = predict * (predict == target)
    // areas of intersection and union
    area_inter, _ = np.histogram(intersection, bins=nbins,
                                 range=(mini, maxi))
    area_pred, _ = np.histogram(predict, bins=nbins,

Instances