Function triplet_loss

script/feature/misc.py:355–369

naive implementation of triplet loss :param criterion: loss function :param f1: [lvl, B, C, H, W] :param f2: [lvl, B, C, H, W] :return: loss

(f1, f2, margin=1.)

    return loss

def triplet_loss(f1, f2, margin=1.):
    ''' 
    naive implementation of triplet loss
    :param criterion: loss function
    :param f1: [lvl, B, C, H, W]
    :param f2: [lvl, B, C, H, W]
    :return:
        loss
    '''
    criterion = nn.TripletMarginLoss(margin=margin, reduction='mean')
    anchor = f1
    positive = f2
    negative = torch.roll(f2, shifts=1, dims=1)
    loss = criterion(anchor, positive, negative)
    return loss

def triplet_loss_hard_negative_mining(f1, f2, margin=1.):
    '''