Function NN_loss

metrics/ChamferDistancePytorch/chamfer_python.py:12–15

(x, y, dim=0)

def NN_loss(x, y, dim=0):
    dist = pairwise_dist(x, y)
    values, indices = dist.min(dim=dim)
    return values.mean()


def distChamfer(a, b):