:mod:`blueoil.metrics.object_keypoint_similarity` ================================================= .. py:module:: blueoil.metrics.object_keypoint_similarity Module Contents --------------- Functions ~~~~~~~~~ .. autoapisummary:: blueoil.metrics.object_keypoint_similarity.compute_object_keypoint_similarity blueoil.metrics.object_keypoint_similarity._compute_oks .. data:: kpt_oks_sigmas .. data:: variances .. function:: compute_object_keypoint_similarity(joints_gt, joints_pred, image_size=(160, 160)) Compute a object keypoint similarity for a batch of examples. :param joints_gt: a numpy array of shape (batch_size, num_joints, 3). :param joints_pred: a numpy array of shape (batch_size, num_joints, 3). :param image_size: a tuple, (height, width). :returns: float. :rtype: oks_batch .. function:: _compute_oks(joints_gt, joints_pred, image_size=(160, 160)) Compute a object keypoint similarity for one example. :param joints_gt: a numpy array of shape (num_joints, 3). :param joints_pred: a numpy array of shape (num_joints, 3). :param image_size: a tuple, (height, width). :returns: float. :rtype: oks