in utils/evaluate.py [0:0]
def eval(self, pred, gt, actions=None):
"""Evaluate
Arguments:
pred {np.ndarray} -- predictions, format (N x 3)
gt {np.ndarray} -- ground truth, format (N x 3)
Keyword Arguments:
action {str} -- action name (default: {None})
"""
for pid, (pose_in, pose_target) in enumerate(zip(pred, gt)):
err = compute_error(pose_in, pose_target)
if actions:
act_name = self._map_action_name(actions[pid])
# add element to dictionary if not there yet
if not self._is_action_stored(act_name):
self._init_action(act_name)
self.error[act_name].append(err)
# add to all
act_name = 'All'
self.error[act_name].append(err)