def METHOD_NAME(self, pred):
    pred = torch.sigmoid(pred)
    n_i, n_c = pred.size()
    bg_score = pred[:, -1].view(n_i, 1)
    if self.test_with_obj:
        pred[:, :-1] *= (1 - bg_score)
    return pred