Function get_pairwise_comp_probs

ADHMR/lib/utils/relation.py:7–15

(batch_preds, batch_std_labels, sigma=None)

import torch.nn.functional as F

def get_pairwise_comp_probs(batch_preds, batch_std_labels, sigma=None):
    batch_s_ij = torch.unsqueeze(batch_preds, dim=2) - torch.unsqueeze(batch_preds, dim=1)
    batch_p_ij = torch.sigmoid(sigma * batch_s_ij)

    batch_std_diffs = torch.unsqueeze(batch_std_labels, dim=2) - torch.unsqueeze(batch_std_labels, dim=1)
    batch_Sij = torch.clamp(batch_std_diffs, min=-1.0, max=1.0)
    batch_std_p_ij = 0.5 * (1.0 + batch_Sij)

    return batch_p_ij, batch_std_p_ij

def rankloss(score, metric,mask=None,sigma=None):
    gtrank = (-metric).argsort().argsort().float()