in utils/evaluate.py [0:0]
def compute_error(pred, gt):
"""Compute error
Arguments:
pred {np.ndarray} -- format (N x 3)
gt {np.ndarray} -- format (N x 3)
Returns:
float -- error
"""
if pred.shape[1] != 3:
pred = np.transpose(pred, [1, 0])
if gt.shape[1] != 3:
gt = np.transpose(gt, [1, 0])
assert pred.shape == gt.shape
error = np.sqrt(np.sum((pred - gt)**2, axis=1))
return np.mean(error)