(aug_scores)
Merge augmented bbox scores.
Merge augmented bbox scores.
Merge augmented bbox scores .
def merge_aug_scores(aug_scores):
    """Merge augmented bbox scores."""
    if isinstance(aug_scores[0], torch.Tensor):
        return torch.mean(torch.stack(aug_scores), dim=0)
    else:
        return np.mean(aug_scores, axis=0)
def merge_aug_scores ( aug_scores ) : if isinstance ( aug_scores [ 0 ] , torch . Tensor ) : return torch . mean ( torch . stack ( aug_scores ) , dim = 0 ) else : return np . mean ( aug_scores , axis = 0 )